Remove autotune package dependency from tests and move fusion examples to standalone package with relative imports

Author: weiT1993

examples/plot_all.py

@@ -2,10 +2,9 @@
 
 import numpy as np
 
-from autotune.analysis.metrics import check_correctness
-from fusion.fusion_chain import FusionChain
-from fusion.operators import Operator
-from fusion.tensors import Tensor
+from .fusion_chain import FusionChain
+from .operators import Operator
+from .tensors import Tensor
 
 
 class Rowmax(Operator):
@@ -328,10 +327,10 @@ def test_flash_attention_fusion() -> None:
     )
 
     result_standard = fusion.execute(fusion_axis="fusion", fusion_step_size=seq_len, input_tensors=input_tensors)
-    check_correctness(golden, result_standard.data, atol, rtol, verbose=True)
+    np.testing.assert_allclose(result_standard.data, golden, atol=atol, rtol=rtol)
 
     result_fused = fusion.execute(fusion_axis="fusion", fusion_step_size=32, input_tensors=input_tensors, verbose=True)
-    check_correctness(golden, result_fused.data, atol, rtol, verbose=True)
+    np.testing.assert_allclose(result_fused.data, golden, atol=atol, rtol=rtol)
 
 
 if __name__ == "__main__":

examples/attention_fusion_numpy.py fusion/attention_fusion_numpy.pyexamples/attention_fusion_numpy.py renamed to fusion/attention_fusion_numpy.py

@@ -2,10 +2,9 @@
 
 import numpy as np
 
-from autotune.analysis.metrics import check_correctness
-from fusion.fusion_chain import FusionChain
-from fusion.operators import Operator
-from fusion.tensors import Tensor
+from .fusion_chain import FusionChain
+from .operators import Operator
+from .tensors import Tensor
 
 
 class SumSquares(Operator):
@@ -214,8 +213,8 @@ def test_rmsnorm_matmul_fusion() -> None:
     result_fused = fusion.execute(fusion_axis="hidden", fusion_step_size=256, input_tensors=input_tensors, verbose=True)
     result_standard = fusion.execute(fusion_axis="hidden", fusion_step_size=hidden_dim, input_tensors=input_tensors)
     golden = rmsnorm_matmul_golden(lhs, rhs, epsilon)
-    check_correctness(golden, result_standard.data, atol, rtol, verbose=True)
-    check_correctness(golden, result_fused.data, atol, rtol, verbose=True)
+    np.testing.assert_allclose(result_standard.data, golden, atol=atol, rtol=rtol)
+    np.testing.assert_allclose(result_fused.data, golden, atol=atol, rtol=rtol)
 
 
 if __name__ == "__main__":