Remove traceable from `_aten_scaled_dot_product_efficient_attention_f…

…illin_empty_outputs` (#1585) Fix _aten_scaled_dot_product_efficient_attention_fillin_empty_outputs regression Co-authored-by: Ti-Tai Wang <[email protected]>
microsoft · Jun 3, 2024 · 1efd8e6 · 1efd8e6
1 parent d39ff45
commit 1efd8e6
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/onnxscript/function_libs/torch_lib/ops/nn.py b/onnxscript/function_libs/torch_lib/ops/nn.py
@@ -1779,7 +1779,7 @@ def aten__scaled_dot_product_flash_attention(
     )
 
 
-@torch_op("aten::_scaled_dot_product_efficient_attention", private=True, traceable=True)
+@torch_op("aten::_scaled_dot_product_efficient_attention", private=True)
 def _aten_scaled_dot_product_efficient_attention_fillin_empty_outputs(
     query: TFloat,
     compute_log_sumexp: bool,