Why there is no Ort::MemoryInfo::CreateGPU or CreateCUDA?
#23356
Unanswered
UnlimitedR
asked this question in
API Q&A
Replies: 1 comment
|
I don't know why they did not define them, but you can use |
0 replies
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Uh oh!
There was an error while loading. Please reload this page.
I see all the codes are like this
Ort::MemoryInfo memoryInfo = Ort::MemoryInfo::CreateCpu(OrtAllocatorType::OrtArenaAllocator, OrtMemType::OrtMemTypeDefault); Ort::Value input_tensor = Ort::Value::CreateTensor<float>(memoryInfo, ...);why it is always
CreateCpu? what if I want a CUDA tensor to accelerate the calculation?and what about
OrtAllocatorType? There seems to beOrtDeviceAllocatorandOrtArenaAllocator, I'm not sure about their differences.All reactions