Fix jetson

triton-inference-server · Jun 21, 2023 · 8e5c545 · 8e5c545
1 parent a629ff1
commit 8e5c545
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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(