[TVMScript][TIR] Add source kernel intetration via call_kernel

apache / tvm

Open deep learning compiler stack for cpu, gpu and specialized accelerators

https://tvm.apache.org/

Apache License 2.0

11.67k stars 3.45k forks source link

Closed vinx13 closed 1 day ago

vinx13 commented 3 days ago

Similar to triton integration, this adds integration with cuda source kernel via call_kernel in TIR

cc @tqchen

tqchen commented 2 days ago

@vinx13 please fix lint