p70865324
/
InfiniTensor
forked from jiuyuan/InfiniTensor
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

fix: 重新导出 cuda_runtime() 

Signed-off-by: YdrMaster <ydrml@hotmail.com>
This commit is contained in:
YdrMaster 2023-02-23 17:20:05 +08:00
parent 9ab78f13f7
commit dc79b72655
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
pyinfinitensor/src/pyinfinitensor/onnx.py
View File

@ -28,6 +28,10 @@ from functools import reduce
cpu_runtime = backend.cpu_runtime() cpu_runtime = backend.cpu_runtime()
def cuda_runtime():
return backend.cuda_runtime()
def from_onnx(model: ModelProto, runtime) -> backend.GraphHandler: def from_onnx(model: ModelProto, runtime) -> backend.GraphHandler:
model = infer_shapes(model) model = infer_shapes(model)
handler = backend.GraphHandler(runtime) handler = backend.GraphHandler(runtime)