From 8e5c5454873867d5dc50ead58e096e445808da35 Mon Sep 17 00:00:00 2001 From: Iman Tabrizian Date: Mon, 12 Jun 2023 11:25:31 -0400 Subject: [PATCH] Fix jetson --- qa/L0_backend_python/python_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/qa/L0_backend_python/python_test.py b/qa/L0_backend_python/python_test.py index b28b0c48b3c..798529225b4 100644 --- a/qa/L0_backend_python/python_test.py +++ b/qa/L0_backend_python/python_test.py @@ -38,7 +38,6 @@ import os from tritonclient.utils import * -import tritonclient.utils.cuda_shared_memory as cuda_shared_memory import tritonclient.http as httpclient TEST_JETSON = bool(int(os.environ.get('TEST_JETSON', 0))) @@ -63,6 +62,7 @@ def _infer_help(self, model_name, shape, data_type): self.assertTrue(np.all(input_data_0 == output0)) def _create_cuda_region(self, client, size, name): + import tritonclient.utils.cuda_shared_memory as cuda_shared_memory shm0_handle = cuda_shared_memory.create_shared_memory_region( name, byte_size=size, device_id=0) client.register_cuda_shared_memory(