std :: ml::gpu_rocm
#include <std/ml/gpu_rocm.h> source
#include <std/ml/gpu_rocm.h>
#define ROCM_FP16 1
same values as tensor_half.h's TENSOR_FP16 / TENSOR_BF16