Skip to content

Commit

Permalink
PR #11557: Fix unittest build error
Browse files Browse the repository at this point in the history
Imported from GitHub PR openxla/xla#11557

This PR fix the build error for tests:

./bazel-6.5.0rc1-linux-x86_64 \
    test \
    --flaky_test_attempts=1 \
    --config=cuda \
    --compilation_mode=dbg \
    --action_env TF_CUDA_COMPUTE_CAPABILITIES=compute_90 \
    --test_summary=detailed \
    --runs_per_test=1 \
    --nocache_test_results \
    --test_output=streamed  \
   //xla/service/gpu/tests:gemm_rewrite_test_gpu

Copybara import of the project:

--
3b4f5cf65eab164b4fa2d79721ee618da772d600 by Shawn Wang <shawnw@nvidia.com>:

fix unittest build error

Merging this change closes #11557

PiperOrigin-RevId: 625217879
  • Loading branch information
shawnwang18 authored and tensorflower-gardener committed Apr 16, 2024
1 parent ab7d6c5 commit f049720
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 1 deletion.
1 change: 1 addition & 0 deletions third_party/xla/xla/backends/interpreter/BUILD
Expand Up @@ -150,6 +150,7 @@ cc_library(
"//xla:status_macros",
"//xla:xla_data_proto_cc",
"//xla/stream_executor",
"//xla/stream_executor:event_interface",
"//xla/stream_executor:stream_executor_interface",
"//xla/stream_executor/host:host_stream",
"@com_google_absl//absl/functional:any_invocable",
Expand Down
1 change: 1 addition & 0 deletions third_party/xla/xla/backends/interpreter/executor.h
Expand Up @@ -31,6 +31,7 @@ limitations under the License.
#include "xla/stream_executor/device_description.h"
#include "xla/stream_executor/device_memory.h"
#include "xla/stream_executor/event.h"
#include "xla/stream_executor/event_interface.h"
#include "xla/stream_executor/host/host_stream.h"
#include "xla/stream_executor/host_memory_allocation.h"
#include "xla/stream_executor/kernel.h"
Expand Down
2 changes: 1 addition & 1 deletion third_party/xla/xla/stream_executor/BUILD
Expand Up @@ -376,7 +376,7 @@ cc_library(

#===--------------------------------------------------------------------------------------------===#
# StreamExecutor internal implementation (visible to StreamExecutor platform implementations)
#===--------------------------------------------------------------------------------------------===#'
#===--------------------------------------------------------------------------------------------===#

# This header-only library is an internal implementation detail of StreamExecutor (to break
# dependency cycles between internal libraries). All StreamExecutor clients should depend on a
Expand Down

0 comments on commit f049720

Please sign in to comment.