[GEMM] [Tuning] Add an option to disable warmup compilation

ROCm / triton

Development repository for the Triton language and compiler

MIT License

80 stars 23 forks source link

[GEMM] [Tuning] Add an option to disable warmup compilation #495

Closed zhanglx13 closed 5 months ago

zhanglx13 commented 5 months ago

For pytorch 1, we cannot pass a dtype to the matmul warmup kernel. This PR adds an option no_warmup to disable the pre-compilation stage of the tuning process.

sunway513 commented 5 months ago

@Cemberk @amathews-amd plz review.