Skip to content

Commit 9fd0f1a

Browse files
committed
Remove tensor load store operations
1 parent 808b378 commit 9fd0f1a

File tree

2 files changed

+0
-39
lines changed

2 files changed

+0
-39
lines changed

mlir/include/mlir/Dialect/AMDGPU/IR/AMDGPU.td

Lines changed: 0 additions & 30 deletions
Original file line numberDiff line numberDiff line change
@@ -1352,34 +1352,4 @@ def AMDGPU_MakeDmaDescriptorOp :
13521352
let hasVerifier = 1;
13531353
}
13541354

1355-
def AMDGPU_TensorLoadToLDSOp :
1356-
AMDGPU_Op<"tensor_load_to_lds", [MemoryEffects<[MemWrite]>, MemoryEffects<[MemRead]>]>,
1357-
Arguments<(ins AMDGPU_TDMDescriptorType: $desc)> {
1358-
let summary = "Load tensors from global memory to LDS.";
1359-
let description = [{
1360-
Load tensors of up to five dimensions from global memory to LDS.
1361-
1362-
The operation is fully described by the descriptor operand.
1363-
}];
1364-
1365-
let assemblyFormat = [{
1366-
$desc attr-dict `:` qualified(type($desc))
1367-
}];
1368-
}
1369-
1370-
def AMDGPU_TensorStoreFromLDSOp :
1371-
AMDGPU_Op<"tensor_store_from_lds", [MemoryEffects<[MemWrite]>, MemoryEffects<[MemRead]>]>,
1372-
Arguments<(ins AMDGPU_TDMDescriptorType: $desc)> {
1373-
let summary = "Store tensors from LDS to global memory.";
1374-
let description = [{
1375-
Store tensors of up to five dimensions from LDS to global memory.
1376-
1377-
The operation is fully described by the descriptor operand.
1378-
}];
1379-
1380-
let assemblyFormat = [{
1381-
$desc attr-dict `:` qualified(type($desc))
1382-
}];
1383-
}
1384-
13851355
#endif // AMDGPU

mlir/test/Dialect/AMDGPU/ops.mlir

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -746,12 +746,3 @@ func.func @make_dma_descriptor(%base: !amdgpu.tdm_base<i32>, %barrier: memref<8x
746746
func.return
747747
}
748748

749-
// CHECK-LABEL: @tensor_load_store
750-
// CHECK-SAME: (%[[DESC:.+]]: !amdgpu.tdm_descriptor)
751-
func.func @tensor_load_store(%desc: !amdgpu.tdm_descriptor) {
752-
// CHECK: amdgpu.tensor_load_to_lds %[[DESC]]
753-
amdgpu.tensor_load_to_lds %desc : !amdgpu.tdm_descriptor
754-
// CHECK: amdgpu.tensor_store_from_lds %[[DESC]]
755-
amdgpu.tensor_store_from_lds %desc : !amdgpu.tdm_descriptor
756-
return
757-
}

0 commit comments

Comments
 (0)