Enable softmax

microsoft · Dec 8, 2023 · 8c5c90d · 8c5c90d
1 parent a0680fe
commit 8c5c90d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mx/mx_mapping.py b/mx/mx_mapping.py
@@ -103,7 +103,7 @@ def addmm_mx(bias, in1, in2, mx_specs=None, name=None):
     "relu6": relu6,
     "sigmoid": sigmoid,
     "silu": silu,
-    # "softmax": softmax, # Causes NaNs?
+    "softmax": softmax, # Causes NaNs?
     "tanh": tanh,
     "add": simd_add,
     "sub": simd_sub,