Skip to content

Commit 46db2e4

Browse files
committed
Update documentation
1 parent 3d62b54 commit 46db2e4

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

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

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1325,12 +1325,12 @@ def AMDGPU_MakeDmaDescriptorOp :
13251325

13261326
```mlir
13271327
// Example of moving a two-dimensional tensor to LDS.
1328-
%base = amdgpu.make_dma_base %lds[0, 0], %global[0, 0] : memref<64x64xi32, #gpu.address_space<workgroup>>, memref<64x64xi32, #gpu.address_space<global>> -> !amdgpu.tdm_base<i32>
1328+
%base = amdgpu.make_dma_base %global[0, 0], %lds[0, 0] : memref<64x64xi32>, memref<64x64xi32, #gpu.address_space<workgroup>> -> !amdgpu.tdm_base<i32>
13291329
%descriptor = amdgpu.make_dma_descriptor %base globalSize [64, 64] globalStride [64, 1] sharedSize [64, 64] : !amdgpu.tdm_base<i32> -> !amdgpu.tdm_descriptor
13301330
amdgpu.tensor_load_to_lds %descriptor : !amdgpu.tdm_descriptor
13311331

13321332
// Example of moving a two dimension tensor to LDS where padding is applied after every integer.
1333-
%base = amdgpu.make_dma_base %lds[0, 0], %global[0, 0] : memref<32x32xi32, #gpu.address_space<workgroup>>, memref<64x64xi32, #gpu.address_space<global>> -> !amdgpu.tdm_base<i32>
1333+
%base = amdgpu.make_dma_base %global[0, 0], %lds[0, 0] : memref<32x32xi32>, memref<64x64xi32, #gpu.address_space<workgroup>> -> !amdgpu.tdm_base<i32>
13341334
%descriptor = amdgpu.make_dma_descriptor %base globalSize [32, 32] globalStride [32, 1] sharedSize [64, 64] padding(%pad pad_every %pad_every) : !amdgpu.tdm_base<i32> -> !amdgpu.tdm_descriptor
13351335
amdgpu.tensor_load_to_lds %descriptor : !amdgpu.tdm_descriptor
13361336
```

0 commit comments

Comments
 (0)