Ran black format and removed the attr_registry_map.h import as it was…

… running into some other issue because of which build was failing
apache · Jul 23, 2024 · 50b44f5 · 50b44f5
1 parent d7c7e22
commit 50b44f5
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 10 deletions.
diff --git a/src/tir/transforms/using_assume_to_reduce_branches.cc b/src/tir/transforms/using_assume_to_reduce_branches.cc
@@ -35,7 +35,6 @@
  *    4. This pass currently works for op_pattern kElemWise and kBroadcast.
  */
 
-#include <tvm/node/attr_registry_map.h>
 #include <tvm/relax/expr.h>
 #include <tvm/relay/op_attr_types.h>
 #include <tvm/tir/builtin.h>

diff --git a/tests/python/relax/test_eliminate_pad_branch_using_buffer_assumption.py b/tests/python/relax/test_eliminate_pad_branch_using_buffer_assumption.py
@@ -631,23 +631,17 @@ def main(
 
 def test_add_primfunc_overcompute():
     add_after = tvm.tir.transform.UseAssumeToReduceBranches()(AddBefore)
-    tvm.ir.structural_equal(
-        add_after["add"], AddExpected["add"], map_free_vars=True
-    )
+    tvm.ir.structural_equal(add_after["add"], AddExpected["add"], map_free_vars=True)
 
 
 def test_sub_primfunc_overcompute():
     sub_after = tvm.tir.transform.UseAssumeToReduceBranches()(SubBefore)
-    tvm.ir.structural_equal(
-        sub_after["sub"], SubExpected["sub"], map_free_vars=True
-    )
+    tvm.ir.structural_equal(sub_after["sub"], SubExpected["sub"], map_free_vars=True)
 
 
 def test_mul_primfunc_overcompute():
     mul_after = tvm.tir.transform.UseAssumeToReduceBranches()(MulBefore)
-    tvm.ir.structural_equal(
-        mul_after["mul"], MulExpected["mul"], map_free_vars=True
-    )
+    tvm.ir.structural_equal(mul_after["mul"], MulExpected["mul"], map_free_vars=True)
 
 
 if __name__ == "__main__":