Skip to content

Commit

Permalink
[stream_executor] NFC: Move stream library into a separate build target
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 574756391
  • Loading branch information
ezhulenev authored and tensorflower-gardener committed Oct 19, 2023
1 parent 3fbde8e commit 2ff1abf
Showing 1 changed file with 23 additions and 4 deletions.
27 changes: 23 additions & 4 deletions third_party/xla/xla/stream_executor/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -449,6 +449,27 @@ cc_library(
],
)

cc_library(
name = "stream",
srcs = ["stream.cc"],
hdrs = ["stream.h"],
visibility = ["//visibility:public"],
deps = [
":blas",
":dnn",
":stream_executor_headers",
":stream_executor_pimpl",
"//xla/stream_executor/platform",
"@com_google_absl//absl/base:core_headers",
"@com_google_absl//absl/functional:any_invocable",
"@com_google_absl//absl/strings",
"@com_google_absl//absl/synchronization",
"@eigen_archive//:eigen3",
"@local_tsl//tsl/platform:logging",
"@local_tsl//tsl/platform:stacktrace",
],
)

cc_library(
name = "temporary_device_memory",
srcs = ["temporary_device_memory.cc"],
Expand Down Expand Up @@ -532,10 +553,7 @@ cc_library(

cc_library(
name = "stream_executor_pimpl",
srcs = [
"stream.cc",
"stream_executor_pimpl.cc",
],
srcs = ["stream_executor_pimpl.cc"],
hdrs = ["stream_executor_pimpl.h"],
visibility = ["//visibility:public"],
deps = [
Expand Down Expand Up @@ -635,6 +653,7 @@ cc_library(
":multi_platform_manager",
":platform",
":scratch_allocator",
":stream",
":stream_executor_headers",
":stream_executor_pimpl",
":temporary_device_memory",
Expand Down

0 comments on commit 2ff1abf

Please sign in to comment.