CUTLASS 3.5.0 (#1411)

2024-03-19 17:51:04 -04:00
parent ffa34e7075
commit 629f4653c3
468 changed files with 48730 additions and 7253 deletions
--- a/test/python/cutlass/evt/evt_mixed_sm80_90.py
+++ b/test/python/cutlass/evt/evt_mixed_sm80_90.py
@ -59,11 +59,11 @@ class TestEVTMixed(EVTTestCaseBase):
            return D, F, F_row_max, E_col_max

        if device_cc() == 80:
-            aligments = [2, 4, 8]
+            alignments = [2, 4, 8]
        else:
            # Sm90 EVT currently only supports 128-bit alignment
-            aligments = [8,]
-        for align in aligments:
+            alignments = [8,]
+        for align in alignments:
            for m, n, k, l in self.get_problem_sizes(align):
                example_inputs = {
                    "accum": self.fake_tensor(self.element, (l, m, n)),