Rate this Page

torch.utils.hipify.hipify_python.replace_extern_shared#

torch.utils.hipify.hipify_python.replace_extern_shared(input_string)[source]#

Match ‘extern __shared__ type foo[];’ syntax and use HIP_DYNAMIC_SHARED() MACRO instead. See: ROCm/hip .. rubric:: Examples

“extern __shared__ char smemChar[];”

=> “HIP_DYNAMIC_SHARED( char, smemChar)”

“extern __shared__ unsigned char smem[];”

=> “HIP_DYNAMIC_SHARED( unsigned char, my_smem)”