Inference on GPU
drvshavva opened this issue · comments
drvshavva commented
Hi, Can I build and use libonnx on gpu ?
jiang jianjun commented
Need write a struct onnx_resolver_t
, and the pass to onnx_context_alloc_from_file function
drvshavva commented
thanks for answer :)
drvshavva commented
hi again, you are saying that we need to write a struct onnx_resolver_t to infer on the gpu. Is it enough ? Is there any other need? And finally how do I change this struct onnx_resolver_t, what structure, what should I do? Many thanks in advance for all the replies :)
jiang jianjun commented
Writing a solver requires a lot of work, you can refer to the default version of the solver implementation