Skip to content

Commit

Permalink
Make cupti_buffer_events_test_cpu compile & link with --config=cuda.
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 675752141
  • Loading branch information
klucke authored and Google-ML-Automation committed Sep 17, 2024
1 parent 3406c60 commit 2a01b6c
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions xla/backends/profiler/gpu/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -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",
],
Expand Down

0 comments on commit 2a01b6c

Please sign in to comment.