diff --git a/build.py b/build.py index b872557c3a..369cad2e41 100755 --- 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))