Tri Dao
|
323c8170bf
|
Support ComputeFn where output type differs from input type (#1771)
This is useful for e.g. function taking in 2 float inputs and turn them to complex
|
2024-09-05 23:25:03 -04:00 |
|
Tri Dao
|
7192f4ab23
|
Add CLayout_64x208 (#1680)
Without this I get compilation error when the extended shapes are enabled
|
2024-08-08 14:00:24 -04:00 |
|
Tri Dao
|
5b283c872c
|
Add more GMMA shapes (#1630)
* Add more GMMA shapes
* Add more shapes for BF16
|
2024-07-29 19:09:51 -04:00 |
|