Skip to content

Commit 068f97c

Browse files
authored
Reapply "fix vector.splat -> vector.broadcast" (#170) (#174)
1 parent ba2e6fe commit 068f97c

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

tests/test_nvgpu_nvvm.py

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -360,7 +360,7 @@ def main(module: any_op_t()):
360360
# CHECK: %[[VAL_89:.*]] = memref.load %[[VAL_44]]{{\[}}%[[VAL_78]], %[[VAL_88]]] : memref<16x16xf16>
361361
# CHECK: %[[VAL_90:.*]] = memref.load %[[VAL_44]]{{\[}}%[[VAL_83]], %[[VAL_86]]] : memref<16x16xf16>
362362
# CHECK: %[[VAL_91:.*]] = memref.load %[[VAL_44]]{{\[}}%[[VAL_83]], %[[VAL_88]]] : memref<16x16xf16>
363-
# CHECK: %[[VAL_92:.*]] = vector.splat %[[VAL_80]] : vector<4x2xf16>
363+
# CHECK: %[[VAL_92:.*]] = vector.broadcast %[[VAL_80]] : f16 to vector<4x2xf16>
364364
# CHECK: %[[VAL_93:.*]] = vector.insert %[[VAL_80]], %[[VAL_92]] [0, 0] : f16 into vector<4x2xf16>
365365
# CHECK: %[[VAL_94:.*]] = vector.insert %[[VAL_82]], %[[VAL_93]] [0, 1] : f16 into vector<4x2xf16>
366366
# CHECK: %[[VAL_95:.*]] = vector.insert %[[VAL_84]], %[[VAL_94]] [1, 0] : f16 into vector<4x2xf16>
@@ -373,7 +373,7 @@ def main(module: any_op_t()):
373373
# CHECK: %[[VAL_102:.*]] = memref.load %[[VAL_45]]{{\[}}%[[VAL_81]], %[[VAL_78]]] : memref<16x8xf16>
374374
# CHECK: %[[VAL_103:.*]] = memref.load %[[VAL_45]]{{\[}}%[[VAL_86]], %[[VAL_78]]] : memref<16x8xf16>
375375
# CHECK: %[[VAL_104:.*]] = memref.load %[[VAL_45]]{{\[}}%[[VAL_88]], %[[VAL_78]]] : memref<16x8xf16>
376-
# CHECK: %[[VAL_105:.*]] = vector.splat %[[VAL_101]] : vector<2x2xf16>
376+
# CHECK: %[[VAL_105:.*]] = vector.broadcast %[[VAL_101]] : f16 to vector<2x2xf16>
377377
# CHECK: %[[VAL_106:.*]] = vector.insert %[[VAL_101]], %[[VAL_105]] [0, 0] : f16 into vector<2x2xf16>
378378
# CHECK: %[[VAL_107:.*]] = vector.insert %[[VAL_102]], %[[VAL_106]] [0, 1] : f16 into vector<2x2xf16>
379379
# CHECK: %[[VAL_108:.*]] = vector.insert %[[VAL_103]], %[[VAL_107]] [1, 0] : f16 into vector<2x2xf16>
@@ -382,7 +382,7 @@ def main(module: any_op_t()):
382382
# CHECK: %[[VAL_111:.*]] = memref.load %[[VAL_46]]{{\[}}%[[VAL_78]], %[[VAL_81]]] : memref<16x8xf16>
383383
# CHECK: %[[VAL_112:.*]] = memref.load %[[VAL_46]]{{\[}}%[[VAL_83]], %[[VAL_79]]] : memref<16x8xf16>
384384
# CHECK: %[[VAL_113:.*]] = memref.load %[[VAL_46]]{{\[}}%[[VAL_83]], %[[VAL_81]]] : memref<16x8xf16>
385-
# CHECK: %[[VAL_114:.*]] = vector.splat %[[VAL_110]] : vector<2x2xf16>
385+
# CHECK: %[[VAL_114:.*]] = vector.broadcast %[[VAL_110]] : f16 to vector<2x2xf16>
386386
# CHECK: %[[VAL_115:.*]] = vector.insert %[[VAL_110]], %[[VAL_114]] [0, 0] : f16 into vector<2x2xf16>
387387
# CHECK: %[[VAL_116:.*]] = vector.insert %[[VAL_111]], %[[VAL_115]] [0, 1] : f16 into vector<2x2xf16>
388388
# CHECK: %[[VAL_117:.*]] = vector.insert %[[VAL_112]], %[[VAL_116]] [1, 0] : f16 into vector<2x2xf16>
@@ -413,6 +413,7 @@ def main(module: any_op_t()):
413413
# CHECK: }
414414
# CHECK: }
415415

416+
print(mod)
416417
filecheck_with_comments(mod)
417418

418419

0 commit comments

Comments
 (0)