diff --git a/xla/backends/profiler/gpu/BUILD b/xla/backends/profiler/gpu/BUILD index 4ae9cd4e775f7..44e71a6f7d1fa 100644 --- a/xla/backends/profiler/gpu/BUILD +++ b/xla/backends/profiler/gpu/BUILD @@ -368,6 +368,9 @@ xla_test( ], deps = [ ":cupti_buffer_events", + ":cupti_collector", + ":cupti_tracer", + ":cupti_utils", "@tsl//tsl/platform:test", "@tsl//tsl/platform:test_main", ],