Use ScopedActivateContext instead of wrapper ScopedActivateExecutorCo…

…ntext in tensorflow/core. PiperOrigin-RevId: 674028097
openxla · Sep 12, 2024 · fa5c1cf · fa5c1cf
1 parent 84ab33a
commit fa5c1cf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/xla/stream_executor/gpu/BUILD b/xla/stream_executor/gpu/BUILD
@@ -57,7 +57,7 @@ package(
         "//xla/pjrt:__subpackages__",
         "//xla/service/gpu:__subpackages__",
         "//xla/stream_executor:__subpackages__",
-        "//tensorflow/core/common_runtime/gpu:__subpackages__",
+        "//tensorflow/core:__subpackages__",
         "//third_party/sparse_conv:__subpackages__",
         "//waymo/ml/compiler/triton:__subpackages__",
     ]),