Disable memory tracker on Jetpack until the library is available (#5882)

triton-inference-server · Jun 1, 2023 · 3d8d517 · 3d8d517
1 parent e0e2c1a
commit 3d8d517
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/build.py b/build.py
@@ -607,6 +607,12 @@ def backend_cmake_args(images, components, be, install_dir, library_paths):
         )
         cargs.append(
             cmake_backend_enable(be, 'TRITON_ENABLE_MEMORY_TRACKER', False))
+    elif target_platform() == 'jetpack':
+        print(
+            "Warning: Detected Jetpack build, backend utility 'device memory tracker' will be disabled as Jetpack doesn't contain required version of the library."
+        )
+        cargs.append(
+            cmake_backend_enable(be, 'TRITON_ENABLE_MEMORY_TRACKER', False))
     elif FLAGS.enable_gpu:
         cargs.append(
             cmake_backend_enable(be, 'TRITON_ENABLE_MEMORY_TRACKER', True))