From dc79b7265506c0d139aa31289657d7a9ddf40ac6 Mon Sep 17 00:00:00 2001 From: YdrMaster Date: Thu, 23 Feb 2023 17:20:05 +0800 Subject: [PATCH] =?UTF-8?q?fix:=20=E9=87=8D=E6=96=B0=E5=AF=BC=E5=87=BA=20c?= =?UTF-8?q?uda=5Fruntime()?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: YdrMaster --- pyinfinitensor/src/pyinfinitensor/onnx.py | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/pyinfinitensor/src/pyinfinitensor/onnx.py b/pyinfinitensor/src/pyinfinitensor/onnx.py index ef7d4de2..f815c037 100644 --- a/pyinfinitensor/src/pyinfinitensor/onnx.py +++ b/pyinfinitensor/src/pyinfinitensor/onnx.py @@ -28,6 +28,10 @@ from functools import reduce cpu_runtime = backend.cpu_runtime() +def cuda_runtime(): + return backend.cuda_runtime() + + def from_onnx(model: ModelProto, runtime) -> backend.GraphHandler: model = infer_shapes(model) handler = backend.GraphHandler(runtime)