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)”
