@@ -411,17 +411,19 @@ def ForallOp : SCF_Op<"forall", [
411
411
tensor<?x?xT> to tensor<?x?xT>
412
412
%sC = tensor.extract_slice %o1[h((%thread_id_1, %thread_id_2))]:
413
413
tensor<?x?xT> to tensor<?x?xT>
414
- %sD = matmul ins(%sA, %sB) outs(%sC)
414
+ %sD = linalg.matmul
415
+ ins(%sA, %sB : tensor<?x?xT>, tensor<?x?xT>)
416
+ outs(%sC : tensor<?x?xT>)
415
417
416
418
%spointwise = subtensor %o2[i((%thread_id_1, %thread_id_2))]:
417
419
tensor<?xT> to tensor<?xT>
418
- %sE = add ins(%spointwise) outs(%sD)
420
+ %sE = linalg. add ins(%spointwise : tensor<?xT> ) outs(%sD : tensor<?xT> )
419
421
420
422
scf.forall.in_parallel {
421
- scf.forall .parallel_insert_slice %sD into %o1[h((%thread_id_1, %thread_id_2))]:
423
+ tensor .parallel_insert_slice %sD into %o1[h((%thread_id_1, %thread_id_2))]:
422
424
tensor<?x?xT> into tensor<?x?xT>
423
425
424
- scf.forall .parallel_insert_slice %spointwise into %o2[i((%thread_id_1, %thread_id_2))]:
426
+ tensor .parallel_insert_slice %spointwise into %o2[i((%thread_id_1, %thread_id_2))]:
425
427
tensor<?xT> into tensor<?xT>
426
428
}
427
429
}
@@ -449,10 +451,12 @@ def ForallOp : SCF_Op<"forall", [
449
451
%sC = tensor.extract_slice %o[%i, %j][%tileSize1, %tileSize2][1, 1]
450
452
: tensor<?x?xT> to tensor<?x?xT>
451
453
452
- %add = map {"arith.addf"} ins(%sA, %sB) outs(%sC)
454
+ %add = linalg.map {"arith.addf"}
455
+ ins(%sA, %sB : tensor<?x?xT>, tensor<?x?xT>)
456
+ outs(%sC : tensor<?x?xT>)
453
457
454
458
scf.forall.in_parallel {
455
- scf.forall .parallel_insert_slice %add into
459
+ tensor .parallel_insert_slice %add into
456
460
%o[%i, %j][%tileSize1, %tileSize2][1, 1]
457
461
: tensor<?x?xT> into tensor<?x?xT>
458
462
}
0 commit comments