search

ROCm / triton

Development repository for the Triton language and compiler
MIT License
80 stars 23 forks source link

[GEMM] [Tuning] Option to try different initialization strategies #486

Closed vgokhale closed 5 months ago

vgokhale commented 5 months ago

Added init strategies:

1) trig_float: element at col i in row j gets value sin(i) 2) hpl: uniform random in [-0.5, 0.5) 3) randn: default, uniform random in [0, 1) 4) zeros: all 0.0s