[Triton-MLIR] Minor fixes to enable fused-softmax and layer-norm tutorials (#835)

2022-11-09 10:18:56 +08:00
parent 2da71b2aaa
commit e517b58d59
4 changed files with 5 additions and 5 deletions
--- a/unittest/Conversion/TritonGPUToLLVM/PtxAsmFormatTest.cpp
+++ b/unittest/Conversion/TritonGPUToLLVM/PtxAsmFormatTest.cpp
@@ -112,8 +112,8 @@ TEST_F(PtxAsmFormatTest, MultiLinePTX) {
  mov(valVal1, constVal);
  mov(valVal1, valVal0);

-  EXPECT_EQ(builder.dump(), "mov $0, 0x1;\r\n"
-                            "mov $1, 0x1;\r\n"
+  EXPECT_EQ(builder.dump(), "mov $0, 0x1;\n\t"
+                            "mov $1, 0x1;\n\t"
                            "mov $1, $0;");

  auto values = builder.getAllMLIRArgs();