Skip to content

Commit 42aeac1

Browse files
hsharshatensorflower-gardener
authored andcommitted
PR #12463: [ROCm] Fix build break in xla/service/gpu/runtime/convolution_thunk.cc
Imported from GitHub PR openxla/xla#12463 Fix build break introduced in db8f9c0cf2bf08b171aec2cbe3da2fcb074a6207 Copybara import of the project: -- 42d5839ffb47362a7f6789ba98ef50b31189dccc by Harsha HS <harsha.havanurshamsundara@amd.com>: [ROCm] Fix build break in `xla/service/gpu/runtime/convolution_thunk.cc` -- fd95e342a4e8de6c8348df3576510584a0020150 by Harsha HS <harsha.havanurshamsundara@amd.com>: Gaurd header using TENSORFLOW_USE_ROCM Merging this change closes #12463 PiperOrigin-RevId: 633851634
1 parent a8a6dd4 commit 42aeac1

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

third_party/xla/xla/service/gpu/runtime/BUILD

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -527,7 +527,7 @@ cc_library(
527527
"@com_google_absl//absl/synchronization",
528528
"@com_google_absl//absl/types:span",
529529
"@local_tsl//tsl/platform:errors",
530-
],
530+
] + if_rocm_is_configured(["//xla/service/gpu:stream_executor_util"]),
531531
)
532532

533533
cc_library(

third_party/xla/xla/service/gpu/runtime/convolution_thunk.cc

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,9 @@ limitations under the License.
2727
#include "absl/synchronization/mutex.h"
2828
#include "absl/types/span.h"
2929
#include "xla/service/buffer_assignment.h"
30+
#if TENSORFLOW_USE_ROCM
31+
#include "xla/service/gpu/stream_executor_util.h"
32+
#endif // TENSORFLOW_USE_ROCM
3033
#include "xla/service/gpu/gpu_conv_runner.h"
3134
#include "xla/service/gpu/runtime/thunk.h"
3235
#include "xla/stream_executor/device_memory.h"

0 commit comments

Comments
 (0)