cutlass icon indicating copy to clipboard operation
cutlass copied to clipboard

Add BF16 dtype mapping for python EVT

Open gau-nernst opened this issue 5 months ago • 3 comments

Fixes #1824

I was thinking of adding a test case for this, but currently the dtype is hard-coded to FP16

https://github.com/NVIDIA/cutlass/blob/44dae8b90ef232ea663727470dfbbe9daff6972d/test/python/cutlass/evt/utils/evt_testbed.py#L206

Would take some refactoring to test multiple dtypes at the same time. Lmk if you want to add any tests, or don't add anything is fine.

@jackkosaian

gau-nernst avatar Sep 20 '24 01:09 gau-nernst