ROCm / hipify_torch

MIT License
18 stars 10 forks source link

Updated with constructs needed for Transformer Engine #42

Closed wenchenvincent closed 1 year ago

wenchenvincent commented 1 year ago

It seems our mapping is not up to date with the rocm hipify project:

The added constructs could be found in the HIP doc and HIPIFY mappings: https://docs.amd.com/bundle/HIP_Supported_CUDA_API_Guide_v5.3/page/CUDA_Runtime_API_functions_supported_by_HIP.html https://github.com/ROCm-Developer-Tools/HIPIFY/blob/amd-staging/src/CUDA2HIP_Runtime_API_types.cpp https://github.com/ROCm-Developer-Tools/HIPIFY/blob/amd-staging/src/CUDA2HIP_Runtime_API_functions.cpp#L218-L219