Skip to content

Commit

Permalink
Resolve merge conflicts
Browse files Browse the repository at this point in the history
  • Loading branch information
jayfurmanek committed Jul 10, 2024
1 parent ce55709 commit 01f6bda
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 37 deletions.
34 changes: 0 additions & 34 deletions third_party/xla/xla/service/gpu/fusions/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -79,12 +79,7 @@ cc_library(
xla_test(
name = "in_place_dynamic_update_slice_mlir_test",
srcs = ["in_place_dynamic_update_slice_mlir_test.cc"],
<<<<<<< HEAD
tags = ["no_rocm"], #TODO(rocm) TEMP
use_gpu = True,
=======
backends = ["gpu"],
>>>>>>> 73722cf2cca19d86967eb49e512918cd566548b3
deps = [
":in_place_dynamic_update_slice_mlir",
":mlir_emitter_test_base",
Expand Down Expand Up @@ -382,12 +377,8 @@ cc_library(
xla_test(
name = "loop_mlir_test",
srcs = ["loop_mlir_test.cc"],
<<<<<<< HEAD
tags = ["no_rocm"],
use_gpu = True,
=======
backends = ["gpu"],
>>>>>>> 73722cf2cca19d86967eb49e512918cd566548b3
deps = [
":loop_mlir",
":mlir_emitter_test_base",
Expand Down Expand Up @@ -436,12 +427,7 @@ cc_library(
xla_test(
name = "scatter_mlir_test",
srcs = ["scatter_mlir_test.cc"],
<<<<<<< HEAD
tags = ["no_rocm"],
use_gpu = True,
=======
backends = ["gpu"],
>>>>>>> 73722cf2cca19d86967eb49e512918cd566548b3
deps = [
":mlir_emitter_test_base",
":scatter_mlir",
Expand Down Expand Up @@ -492,12 +478,7 @@ cc_library(
xla_test(
name = "transpose_mlir_test",
srcs = ["transpose_mlir_test.cc"],
<<<<<<< HEAD
tags = ["no_rocm"], #TODO(rocm) TEMP
use_gpu = True,
=======
backends = ["gpu"],
>>>>>>> 73722cf2cca19d86967eb49e512918cd566548b3
deps = [
":mlir_emitter_test_base",
":transpose_mlir",
Expand Down Expand Up @@ -887,12 +868,7 @@ cc_library(
xla_test(
name = "reduction_mlir_test",
srcs = ["reduction_mlir_test.cc"],
<<<<<<< HEAD
tags = ["no_rocm"], #TODO(rocm) TEMP
use_gpu = True,
=======
backends = ["gpu"],
>>>>>>> 73722cf2cca19d86967eb49e512918cd566548b3
deps = [
":mlir_emitter_test_base",
":reduction_mlir",
Expand Down Expand Up @@ -977,12 +953,7 @@ cc_library(
xla_test(
name = "concatenate_mlir_test",
srcs = ["concatenate_mlir_test.cc"],
<<<<<<< HEAD
tags = ["no_rocm"], #TODO(rocm) TEMP
use_gpu = True,
=======
backends = ["gpu"],
>>>>>>> 73722cf2cca19d86967eb49e512918cd566548b3
deps = [
":concatenate_mlir",
":mlir_emitter_test_base",
Expand Down Expand Up @@ -1114,12 +1085,7 @@ cc_library(
xla_test(
name = "input_slices_mlir_test",
srcs = ["input_slices_mlir_test.cc"],
<<<<<<< HEAD
tags = ["no_rocm"], #TODO(rocm) TEMP
use_gpu = True,
=======
backends = ["gpu"],
>>>>>>> 73722cf2cca19d86967eb49e512918cd566548b3
deps = [
":input_slices_mlir",
":mlir_emitter_test_base",
Expand Down
4 changes: 1 addition & 3 deletions third_party/xla/xla/service/gpu/gpu_compiler_test.cc
Original file line number Diff line number Diff line change
Expand Up @@ -343,18 +343,16 @@ ENTRY main {

TEST_F(GpuCompilerTest,
GemmFusionIsNoOpWhenGemmFusionAutotunerFallsBackToCublas) {
<<<<<<< HEAD
if (std::holds_alternative<se::RocmComputeCapability>(GpuComputeComp())) {
GTEST_SKIP() << "Folder structure differences prevents finding of gpu_compiler_test_autotune_db.textproto.";
=======
}
auto cc = backend()
.default_stream_executor()
->GetDeviceDescription()
.cuda_compute_capability();
if (!cc.IsAtLeastAmpere()) {
GTEST_SKIP() << "Autotuning results have only been generated for Ampere "
<< "and Hopper GPUs";
>>>>>>> 73722cf2cca19d86967eb49e512918cd566548b3
}
const absl::string_view hlo_string = R"(
HloModule test
Expand Down

0 comments on commit 01f6bda

Please sign in to comment.