search

ROCm / triton

Development repository for the Triton language and compiler
MIT License
92 stars 29 forks source link

gemm tuning script to support gemm fp8/f16 mixed input #342

Open scxiao opened 1 year ago

scxiao commented 1 year ago

With the fp8/fp16 mixed input for GEMM enabled, the tuning scripts should be changed according to support that. This PR is for that.

zhanglx13 commented 1 year ago

@scxiao Do you mind if I merge https://github.com/ROCmSoftwarePlatform/triton/pull/344 first while you are working on supporting mixed input in the GEMM tuning script?

scxiao commented 1 year ago

@scxiao Do you mind if I merge #344 first while you are working on supporting mixed input in the GEMM tuning script?

As long as there is no performance regression, I am fine with that. Thanks

scxiao commented 1 year ago

Hi @zhanglx13, when you get a chance, could you please review this PR? Thanks