torch.utils.hipify.hipify_python.replace_extern_shared

torch.utils.hipify.hipify_python.replace_extern_shared(input_string) [исходный код]

Сопоставляет синтаксис «extern __shared__ type foo[];» и вместо него использует макрос HIP_DYNAMIC_SHARED(). См.: ROCm/hip .. рубрика:: Примеры

“extern __shared__ char smemChar[];”

=> “HIP_DYNAMIC_SHARED( char, smemChar)”

“extern __shared__ unsigned char smem[];”

=> “HIP_DYNAMIC_SHARED( unsigned char, my_smem)”