fixed errors in PR comment

microsoft · Aug 15, 2024 · 03ba9e9 · 03ba9e9
1 parent 2771a40
commit 03ba9e9
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 11 deletions.
diff --git a/onnxscript/rewriter/onnxruntime/transformers/__init__.py b/onnxscript/rewriter/onnxruntime/transformers/__init__.py
@@ -11,19 +11,19 @@
 )
 
 TRANSFORMERS_FUNCTION_REWRITE_RULES: list[type[function_rule.FunctionRewriteRule]] = [
-    # multihead_attention.GQALlama2RewriteRule,
-    # multihead_attention.GQALlamaSdpa2RewriteRule,
-    # multihead_attention.AttnPhi15RewriteRule,
-    # multihead_attention.MHAStableDiffusionUnetRewriteRule,
+    multihead_attention.GQALlama2RewriteRule,
+    multihead_attention.GQALlamaSdpa2RewriteRule,
+    multihead_attention.AttnPhi15RewriteRule,
+    multihead_attention.MHAStableDiffusionUnetRewriteRule,
     layernorm.LNRewriteRule,
-    # fastgelu.GeluRewriteRule,
-    # biassplitgelu.GegluRewriteRule,
-    # multihead_attention.MHALlama2RewriteRule,
+    fastgelu.GeluRewriteRule,
+    biassplitgelu.GegluRewriteRule,
+    multihead_attention.MHALlama2RewriteRule,
     multihead_attention.GQALlama3RewriteRule,
-    # multihead_attention.AttentionRewriteRule,
+    multihead_attention.AttentionRewriteRule,
     multihead_attention.MLP3RewriteRule,
     multihead_attention.GQALlama3RewriteRuleFirstAttention,
-    # multihead_attention.MLPRewriteRule,
-    # multihead_attention.GQALlamaRewriteRule,
+    multihead_attention.MLPRewriteRule,
+    multihead_attention.GQALlamaRewriteRule,
 
 ]
diff --git a/testingg.py b/testingg.py
@@ -4,7 +4,7 @@
 import onnxscript.optimizer
 import onnxscript.rewriter
 from onnxscript.rewriter import onnxruntime as ort_rewriter
-from onnxscript.ir import convenience as ir_convenience
+
 
 def export_to_onnx(input_model_path: str, output_model_path: str):
     """