From c05f038053c3e70f29b25dfa3aac4a8552251f61 Mon Sep 17 00:00:00 2001 From: GuanLuo Date: Wed, 31 May 2023 15:24:54 -0700 Subject: [PATCH] Disable memory tracker on Jetpack until the library is available --- build.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/build.py b/build.py index 4b6e83fcb4..f6f8d418fc 100755 --- a/build.py +++ b/build.py @@ -606,6 +606,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))