Skip to content

Commit

Permalink
* Basic cc test for collective_permute_cycle_decomposer_test because …
Browse files Browse the repository at this point in the history
…the test is quite basic.

* gunit_main instead of /tsl/platform:test_main because /tsl/platform:test_main doesn't work with VLOG and is not required for this test.

PiperOrigin-RevId: 712643227
  • Loading branch information
toli-y authored and Google-ML-Automation committed Jan 6, 2025
1 parent 8849c58 commit ba9cd83
Show file tree
Hide file tree
Showing 2 changed files with 8 additions and 8 deletions.
8 changes: 4 additions & 4 deletions xla/service/gpu/transforms/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -514,13 +514,13 @@ xla_cc_test(
deps = [
":collective_permute_cycle_decomposer",
"//xla/hlo/ir:hlo",
"//xla/tests:filecheck",
"//xla/tests:hlo_test_base",
"//xla/hlo/testlib:filecheck",
"//xla/hlo/testlib:hlo_hardware_independent_test_base",
"//xla/tsl/lib/core:status_test_util",
"//xla/tsl/platform:statusor",
"@com_google_absl//absl/strings:string_view",
"@com_google_googletest//:gtest",
"@tsl//tsl/platform:statusor",
"@tsl//tsl/platform:test_main",
"@com_google_googletest//:gtest_main",
],
)

Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -25,16 +25,16 @@ limitations under the License.
#include "xla/hlo/ir/hlo_instruction.h"
#include "xla/hlo/ir/hlo_instructions.h"
#include "xla/hlo/ir/hlo_module.h"
#include "xla/tests/filecheck.h"
#include "xla/tests/hlo_test_base.h"
#include "xla/hlo/testlib/filecheck.h"
#include "xla/hlo/testlib/hlo_hardware_independent_test_base.h"
#include "xla/tsl/lib/core/status_test_util.h"
#include "tsl/platform/statusor.h"
#include "xla/tsl/platform/statusor.h"

namespace xla {
namespace {

using ::testing::HasSubstr;
using CollectivePermuteCycleDecomposerTest = HloTestBase;
using CollectivePermuteCycleDecomposerTest = HloHardwareIndependentTestBase;
using Decomposer = CollectivePermuteCycleDecomposer;

HloPrintOptions PrintOptions() {
Expand Down

0 comments on commit ba9cd83

Please sign in to comment.