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

..

cute

Use CUDA runtime API to retrieve function pointer to driver API (#1700)

2024-08-19 13:26:09 -04:00

cutlass

Support ComputeFn where output type differs from input type (#1771)

2024-09-05 23:25:03 -04:00