add gpu op profiles

EnzymeAD · May 3, 2024 · 6909289 · 6909289
1 parent 3157b2f
commit 6909289
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/deps/ReactantExtra/BUILD b/deps/ReactantExtra/BUILD
@@ -196,7 +196,8 @@ cc_library(
         "@xla//xla/stream_executor/cuda:cuda_platform",
         "@xla//xla/stream_executor/cuda:cudnn_plugin",
         "@xla//xla/stream_executor/cuda:cufft_plugin",
-        "@xla//xla/stream_executor:cuda_platform",
+        "@xla//xla/stream_executor:cuda_platform",        
+        "@xla//xla/service/gpu/model:hlo_op_profiles",
 
         "@xla//xla/pjrt:status_casters",
         "@xla//xla/python/ifrt:ifrt",