pytorch
2abdb7d3 - [jiterator] De-template launch_jitted_reduce_kernel (#80968)

Commit
3 years ago
[jiterator] De-template launch_jitted_reduce_kernel (#80968) As with `jitted_gpu_kernel_impl`, this 1. Hoists static variables out and into a parent funciton 2. Moves template arguments into the `jit::KernelDescriptor` struct, as well as changing `vt0` to just be a runtime argument 3. Changes the types of pass-through arguments to `void*` Pull Request resolved: https://github.com/pytorch/pytorch/pull/80968 Approved by: https://github.com/mehtanirav
Author
Committer
Parents
Loading