search

triton-lang / triton

Development repository for the Triton language and compiler
https://triton-lang.org/
MIT License
13.09k stars 1.6k forks source link

[TESTS] Add tests for `matmul` with `out_dtype=float16` #2260

Open jon-chuang opened 1 year ago

jon-chuang commented 1 year ago

Currently, it is always None which defaults to float32

https://github.com/openai/triton/blob/f21b36c8c54f35a88e96d7217e2c6bc9cc02ee69/python/test/unit/operators/test_matmul.py#L179

I believe Hopper tests do cover it but it should be tested for non-Hopper as well.

https://github.com/openai/triton/blob/f21b36c8c54f35a88e96d7217e2c6bc9cc02ee69/python/test/unit/hopper/test_gemm.py#L223

ThomasRaoux commented 1 year ago

It is tested here: https://github.com/openai/triton/blob/main/python/test/unit/language/test_core.py#L2200