onnxruntime
ca7116ca - CUDA EP's ResizeImpl now uses functors, hipify for ROCm EP (#9466)

Commit
4 years ago
CUDA EP's ResizeImpl now uses functors, hipify for ROCm EP (#9466) Support for device function pointers is not yet available for ROCm. Instead, the device function pointers were converted to device functors. Case statements, lambdas, and macros are used for dispatch; as a result, all combinations of kernels are compiled with inlined functors. The basis of this approach can be found in PyTorch. Lastly, hipify and register Resize and Upsample for ROCm EP.
Author
Parents
Loading