add more folding testcases and fix store fp32 folding result to double

apache · Aug 28, 2022 · 74aaa1c · 74aaa1c
1 parent bcf0ac2
commit 74aaa1c
Show file tree

Hide file tree

Showing 2 changed files with 387 additions and 20 deletions.
diff --git a/python/tvm/script/tir/intrin.py b/python/tvm/script/tir/intrin.py
@@ -85,6 +85,11 @@ def truncmod(x, y, span):
     return tvm.tir.truncmod(x, y, span)
 
 
+@register
+def truncdiv(x, y, span):
+    return tvm.tir.truncdiv(x, y, span)
+
+
 @register
 def ceildiv(x, y, span):
     return tvm.tir.ceildiv(x, y, span)