From 5ac38e7103c1adface843632a532a0f1fc59c561 Mon Sep 17 00:00:00 2001 From: Sasha Rush Date: Thu, 21 Mar 2024 20:33:45 -0400 Subject: [PATCH] [Interpreter] Add exp and log to the interpreter. (#3432) Getting errors with these in the latest version of triton. --- python/triton/runtime/interpreter.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/python/triton/runtime/interpreter.py b/python/triton/runtime/interpreter.py index 09141e82267e..6f6c6a85d274 100644 --- a/python/triton/runtime/interpreter.py +++ b/python/triton/runtime/interpreter.py @@ -687,6 +687,8 @@ def _patch_lang_math(lang): "abs": np.abs, "acos": np.arccos, "asin": np.arcsin, + "exp": np.exp, + "log": np.log, "exp2": np.exp2, "log2": np.log2, "max": np.maximum,