static Ort::MemoryInfo MemoryInfo(const TensorDesc& desc) {
const char* device_name = desc.device.is_host() ? "Cpu" : "Rocm";
//const char* device_name = "Cpu";
//printf("device_name = %s\n", device_name);
Ort::MemoryInfo memory_info(device_name, OrtDeviceAllocator, desc.device.device_id(),
OrtMemTypeDefault);
return memory_info;
}
海光的 onnxruntime 中 device_name 应该是什么?我是试过 Rocm 会报错。