Weixing Zhang 34b89a1173 [OPTIMIZER] Tweak warpsPerCTA based on the shape of MMA output (#2581) ...

In current implementation, warpsPerCTA is always set to [numWarps, 1]
for 2 tt.dot fusion scenario. But, it is not optimal for cases such that
tt.dot doesn't have enough parallelism on row dimension but on column
dimension.

2023-11-03 16:40:03 -04:00

..

test_cast_matmul.py

Reformat Python code with yapf. (#2589)

2023-11-02 20:44:17 -07:00

test_functional_regressions.py

Reformat Python code with yapf. (#2589)

2023-11-02 20:44:17 -07:00

test_performance.py

[OPTIMIZER] Tweak warpsPerCTA based on the shape of MMA output (#2581)

2023-11-03 16:40:03 -04:00