Skip to content

Commit 8580ff8

Browse files
loisloGoogle-ML-Automation
authored andcommitted
Give meaningful names to HLO modules in triton_fusion_emitter_int4_device_test.cc.
This will make it easier to identify failing tests among the dumps. PiperOrigin-RevId: 715372509
1 parent be9504c commit 8580ff8

File tree

1 file changed

+16
-16
lines changed

1 file changed

+16
-16
lines changed

xla/service/gpu/fusions/triton/triton_fusion_emitter_int4_device_test.cc

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -98,7 +98,7 @@ class PlainInt4ToPackedInt4RewritePassTest : public TritonTest {
9898
TEST_F(PlainInt4ToPackedInt4RewritePassTest,
9999
DotWithI4WeightsOnLhsWithNonStandardLayoutAndMultplyInEpilogue) {
100100
constexpr absl::string_view kHloText = R"(
101-
HloModule hlo
101+
HloModule DotWithI4WeightsOnLhsWithNonStandardLayoutAndMultplyInEpilogue
102102
103103
fusion {
104104
p_0 = s4[1,128,32]{1,2,0:E(4)} parameter(0)
@@ -383,7 +383,7 @@ TEST_F(TritonTest, NonstandardLayoutInt4) {
383383
TEST_F(TritonTest, NonstandardLayoutWithManyNonContractingDims) {
384384
// We cannot do triton_gemm and we use cuBLAS instead.
385385
constexpr absl::string_view kHloText = R"(
386-
HloModule t
386+
HloModule NonstandardLayoutWithManyNonContractingDims
387387
388388
ENTRY main {
389389
p0 = s4[128,64,192]{1,0,2} parameter(0)
@@ -402,7 +402,7 @@ TEST_F(TritonTest, NonstandardLayoutWithManyNonContractingDims) {
402402
TEST_F(TritonTest, NonstandardLayoutWithManyNonContractingDimsReversedLayout) {
403403
// We cannot do triton_gemm and we use cuBLAS instead.
404404
constexpr absl::string_view kHloText = R"(
405-
HloModule t
405+
HloModule NonstandardLayoutWithManyNonContractingDimsReversedLayout
406406
407407
ENTRY main {
408408
lhs = s4[128,64,192]{0,1,2} parameter(0)
@@ -420,7 +420,7 @@ TEST_F(TritonTest, NonstandardLayoutWithManyNonContractingDimsReversedLayout) {
420420

421421
TEST_F(TritonTest, NegatePlusConvertHLO) {
422422
constexpr absl::string_view kHloText = R"(
423-
HloModule t
423+
HloModule NegatePlusConvertHLO
424424
425425
ENTRY main {
426426
lhs = s4[16,32,64]{2,1,0} parameter(0)
@@ -440,7 +440,7 @@ TEST_F(TritonTest, NegatePlusConvertHLO) {
440440

441441
TEST_F(TritonTest, RejectTritonFusionForWithMinorBatchDim) {
442442
constexpr absl::string_view kHloText = R"(
443-
HloModule t
443+
HloModule RejectTritonFusionForWithMinorBatchDim
444444
445445
ENTRY main {
446446
lhs = s4[32,64,16]{2,1,0} parameter(0)
@@ -465,7 +465,7 @@ TEST_F(TritonTest, LHSWithMinorDimEqualTo1) {
465465
// We prove that triton can handle int4 dot with non contracting dim size
466466
// equal to 1.
467467
constexpr absl::string_view kHloText = R"(
468-
HloModule t
468+
HloModule LHSWithMinorDimEqualTo1
469469
470470
triton_computation {
471471
lhs = s4[16,1024,1]{2,1,0} parameter(0)
@@ -494,7 +494,7 @@ TEST_F(TritonTest, RHSWithMinorDimEqualTo1) {
494494
// We prove that triton can handle int4 dot with non contracting dim size
495495
// equal to 1.
496496
constexpr absl::string_view kHloText = R"(
497-
HloModule t
497+
HloModule RHSWithMinorDimEqualTo1
498498
499499
triton_computation {
500500
lhs = bf16[16,1024,64]{2,1,0} parameter(0)
@@ -524,7 +524,7 @@ TEST_F(TritonTest, LHSNonMinorContractingDim) {
524524
// We prove that triton can handle int4 dot with non minor
525525
// lhs_contracting_dim.
526526
constexpr absl::string_view kHloText = R"(
527-
HloModule t
527+
HloModule LHSNonMinorContractingDim
528528
529529
triton_computation {
530530
lhs = s4[1024,8]{1,0} parameter(0)
@@ -552,7 +552,7 @@ TEST_F(TritonTest, LHSNonMinorContractingDimWithBatchDim0) {
552552
// We prove that triton can handle int4 dot with non minor
553553
// lhs_contracting_dim.
554554
constexpr absl::string_view kHloText = R"(
555-
HloModule t
555+
HloModule LHSNonMinorContractingDimWithBatchDim0
556556
557557
triton_computation {
558558
lhs = s4[16,1024,8]{2,1,0} parameter(0)
@@ -580,7 +580,7 @@ TEST_F(TritonTest, LHSNonMinorContractingDimWithBatchDim0) {
580580
TEST_F(TritonTest, LHSMinorContractingDim) {
581581
// We prove that triton can handle int4 dot with minor lhs_contracting_dim.
582582
constexpr absl::string_view kHloText = R"(
583-
HloModule t
583+
HloModule LHSMinorContractingDim
584584
585585
triton_computation {
586586
lhs = s4[8,1024]{1,0} parameter(0)
@@ -604,7 +604,7 @@ TEST_F(TritonTest, LHSMinorContractingDim) {
604604

605605
TEST_F(TritonTest, ConvertPlusNegate) {
606606
constexpr absl::string_view kHloText = R"(
607-
HloModule t
607+
HloModule ConvertPlusNegate
608608
609609
triton_computation {
610610
lhs = s4[8,1024]{1,0} parameter(0)
@@ -630,7 +630,7 @@ TEST_F(TritonTest, ConvertPlusNegate) {
630630
TEST_F(TritonTest, LHSMinorContractingDimWithBatchDim0) {
631631
// We prove that triton can handle int4 dot with minor lhs_contracting_dim.
632632
constexpr absl::string_view kHloText = R"(
633-
HloModule t
633+
HloModule LHSMinorContractingDimWithBatchDim0
634634
635635
triton_computation {
636636
lhs = s4[16,8,1024]{2,1,0} parameter(0)
@@ -657,7 +657,7 @@ TEST_F(TritonTest, LHSMinorContractingDimWithBatchDim0) {
657657

658658
TEST_F(TritonTest, RHSTestWithNotMinorContractingDim) {
659659
constexpr absl::string_view kHloText = R"(
660-
HloModule t
660+
HloModule RHSTestWithNotMinorContractingDim
661661
662662
triton_computation {
663663
lhs = bf16[8,1024]{1,0} parameter(0)
@@ -682,7 +682,7 @@ TEST_F(TritonTest, RHSTestWithNotMinorContractingDim) {
682682

683683
TEST_F(TritonTest, RHSTestWithMinorContractingDim) {
684684
constexpr absl::string_view kHloText = R"(
685-
HloModule t
685+
HloModule RHSTestWithMinorContractingDim
686686
687687
triton_computation {
688688
lhs = bf16[8,1024]{1,0} parameter(0)
@@ -707,7 +707,7 @@ TEST_F(TritonTest, RHSTestWithMinorContractingDim) {
707707

708708
TEST_F(TritonTest, RHSTestWithMinorContractingDimWithBatchDim) {
709709
constexpr absl::string_view kHloText = R"(
710-
HloModule t
710+
HloModule RHSTestWithMinorContractingDimWithBatchDim
711711
712712
triton_computation {
713713
lhs = bf16[16,8,1024]{2,1,0} parameter(0)
@@ -734,7 +734,7 @@ TEST_F(TritonTest, RHSTestWithMinorContractingDimWithBatchDim) {
734734

735735
TEST_F(TritonTest, RHSTestWithNotMinorContractingDimWithBatchDim0) {
736736
constexpr absl::string_view kHloText = R"(
737-
HloModule t
737+
HloModule RHSTestWithNotMinorContractingDimWithBatchDim0
738738
739739
triton_computation {
740740
lhs = bf16[16,8,1024]{2,1,0} parameter(0)

0 commit comments

Comments
 (0)