derek.holloway
7502bf64a4
CodeQL / Analyze (actions) (push) Canceled after 0s
CodeQL / Analyze (python) (push) Canceled after 0s
pre-commit / pre-commit (push) Canceled after 0s
Multi-Arch CI / setup (push) Canceled after 0s
Publish build_manylinux_x86_64 images / publish_build_manylinux_x86_64 (push) Canceled after 0s
Setup (Multi-Arch) / setup (push) Canceled after 0s
Multi-Arch CI / Linux::${{ fromJSON(needs.setup.outputs.linux_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubi10 image / publish_no_rocm_image_ubi10 (push) Canceled after 0s
Multi-Arch CI / Windows::${{ fromJSON(needs.setup.outputs.windows_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Multi-Arch CI / Manifest Diff (push) Canceled after 0s
TheRock Manifest Diff Report / Generate Manifest Diff Report (push) Canceled after 0s
Multi-Arch CI / CI Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_ocl_rt images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_rocgdb images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Unit Tests / Unit Tests :: ubuntu-24.04 (push) Canceled after 0s
Unit Tests / Unit Tests :: windows-2022 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_media (push) Canceled after 0s
Unit Tests / Unit Tests Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_openmpi (push) Canceled after 0s
40 lines
1.7 KiB
MLIR
40 lines
1.7 KiB
MLIR
// RUN: mlir-opt --canonicalize --split-input-file %s | FileCheck %s
|
|
|
|
// CHECK-LABEL: @make_dma_descriptor_fold
|
|
// CHECK-SAME: (%[[BASE:.+]]: !amdgpu.tdm_base<i32>, %[[IDX:.+]]: index, %[[I32:.+]]: i32)
|
|
func.func @make_dma_descriptor_fold(%base: !amdgpu.tdm_base<i32>, %idx: index, %i32: i32) -> !amdgpu.tdm_descriptor {
|
|
%c64 = arith.constant 64 : index
|
|
|
|
// CHECK: amdgpu.make_dma_descriptor %[[BASE]]
|
|
%0 = amdgpu.make_dma_descriptor %base
|
|
// CHECK-SAME: globalSize [64, 64]
|
|
globalSize [%c64, %c64]
|
|
// CHECK-SAME: globalStride [64, 1]
|
|
globalStride [%c64, 1]
|
|
// CHECK-SAME: sharedSize [64, 64]
|
|
sharedSize [%c64, %c64]
|
|
iterate %idx, %i32, %idx
|
|
: !amdgpu.tdm_base<i32> -> !amdgpu.tdm_descriptor
|
|
func.return %0 : !amdgpu.tdm_descriptor
|
|
}
|
|
|
|
// -----
|
|
|
|
// CHECK-LABEL: @make_gather_dma_descriptor_fold
|
|
// CHECK-SAME: (%[[BASE:.+]]: !amdgpu.tdm_gather_base<i32, i32>, %[[IDX:.+]]: index, %[[I32:.+]]: i32, %[[INDICES:.+]]: vector<8xi32>)
|
|
func.func @make_gather_dma_descriptor_fold(%base: !amdgpu.tdm_gather_base<i32, i32>, %idx: index, %i32: i32, %indices: vector<8xi32>) -> !amdgpu.tdm_descriptor {
|
|
%c64 = arith.constant 64 : index
|
|
|
|
// CHECK: amdgpu.make_gather_dma_descriptor %[[BASE]][%[[INDICES]]]
|
|
%0 = amdgpu.make_gather_dma_descriptor %base [%indices]
|
|
// CHECK-SAME: globalSize [64, 64]
|
|
globalSize [%c64, %c64]
|
|
// CHECK-SAME: globalStride [64, 1]
|
|
globalStride [%c64, 1]
|
|
// CHECK-SAME: sharedSize [64, 64]
|
|
sharedSize [%c64, %c64]
|
|
iterate %idx, %i32, %idx
|
|
: !amdgpu.tdm_gather_base<i32, i32>, vector<8xi32> -> !amdgpu.tdm_descriptor
|
|
func.return %0 : !amdgpu.tdm_descriptor
|
|
}
|