saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-06 00:03:22 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

undo hipify of __half to rocblas_half (#18573)

Browse source 
Fixes build issue seen with newer ROCm releases

Co-authored-by: Jeff Daily <jeff.daily@amd.com>
This commit is contained in:
Ted Themistokleous 2023-11-24 05:04:23 -05:00 committed by GitHub
parent b9c935f605
commit 7b2aefa856
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
tools/ci_build/amd_hipify.py
View file

@ -35,6 +35,9 @@ def hipify(hipify_perl_path, src_file_path, dst_file_path):
s = s.replace("HIPBLAS_OP_T", "rocblas_operation_transpose")
s = s.replace("HIPBLAS_OP_N", "rocblas_operation_none")
# in rocm 6.0, hipify-perl, the -roc option also maps __half -> rocblas_half which we don't want
s = s.replace("rocblas_half", "__half")
s = s.replace("RegisterCudaContribKernels", "RegisterRocmContribKernels")
s = s.replace("cudaEvent", "hipEvent")
s = s.replace("CreateCudaAllocator", "CreateRocmAllocator")