@@ -98,7 +98,7 @@ class PlainInt4ToPackedInt4RewritePassTest : public TritonTest {
9898TEST_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) {
383383TEST_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) {
402402TEST_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
421421TEST_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
441441TEST_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) {
580580TEST_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
605605TEST_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) {
630630TEST_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
658658TEST_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
683683TEST_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
708708TEST_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
735735TEST_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