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
122 lines
6.3 KiB
MLIR
122 lines
6.3 KiB
MLIR
// RUN: mlir-opt --canonicalize %s | FileCheck %s
|
|
|
|
/// Check `ptr_add` canonicalizer patterns.
|
|
|
|
// CHECK-LABEL: @zero_offset
|
|
// CHECK-SAME: (%[[PTR_0:.*]]: !ptr.ptr<#ptr.generic_space>)
|
|
func.func @zero_offset(%ptr: !ptr.ptr<#ptr.generic_space>) -> !ptr.ptr<#ptr.generic_space> {
|
|
// CHECK-NOT: index.constant
|
|
// CHECK-NOT: ptr.ptr_add
|
|
// CHECK: return %[[PTR_0]] : !ptr.ptr<#ptr.generic_space>
|
|
// CHECK: }
|
|
%off = index.constant 0
|
|
%res0 = ptr.ptr_add %ptr, %off : !ptr.ptr<#ptr.generic_space>, index
|
|
return %res0 : !ptr.ptr<#ptr.generic_space>
|
|
}
|
|
|
|
/// Tests the the `from_ptr` folder.
|
|
// CHECK-LABEL: @test_from_ptr_0
|
|
// CHECK-SAME: (%[[MEM_REF:.*]]: memref<f32, #ptr.generic_space>)
|
|
func.func @test_from_ptr_0(%mr: memref<f32, #ptr.generic_space>) -> memref<f32, #ptr.generic_space> {
|
|
// CHECK-NOT: ptr.to_ptr
|
|
// CHECK-NOT: ptr.get_metadata
|
|
// CHECK-NOT: ptr.from_ptr
|
|
// CHECK: return %[[MEM_REF]]
|
|
%ptr = ptr.to_ptr %mr : memref<f32, #ptr.generic_space> -> !ptr.ptr<#ptr.generic_space>
|
|
%mda = ptr.get_metadata %mr : memref<f32, #ptr.generic_space>
|
|
%res = ptr.from_ptr %ptr metadata %mda : !ptr.ptr<#ptr.generic_space> -> memref<f32, #ptr.generic_space>
|
|
return %res : memref<f32, #ptr.generic_space>
|
|
}
|
|
|
|
/// Check the op doesn't fold because folding a ptr-type with metadata requires knowing the origin of the metadata.
|
|
// CHECK-LABEL: @test_from_ptr_1
|
|
// CHECK-SAME: (%[[MEM_REF:.*]]: memref<f32, #ptr.generic_space>)
|
|
func.func @test_from_ptr_1(%mr: memref<f32, #ptr.generic_space>) -> memref<f32, #ptr.generic_space> {
|
|
// CHECK: ptr.to_ptr
|
|
// CHECK: ptr.from_ptr
|
|
%ptr = ptr.to_ptr %mr : memref<f32, #ptr.generic_space> -> !ptr.ptr<#ptr.generic_space>
|
|
%res = ptr.from_ptr %ptr : !ptr.ptr<#ptr.generic_space> -> memref<f32, #ptr.generic_space>
|
|
return %res : memref<f32, #ptr.generic_space>
|
|
}
|
|
|
|
/// Check that the ops cannot be folded because the metadata cannot be guaranteed to be the same.
|
|
// CHECK-LABEL: @test_from_ptr_2
|
|
func.func @test_from_ptr_2(%mr: memref<f32, #ptr.generic_space>, %md: !ptr.ptr_metadata<memref<f32, #ptr.generic_space>>) -> memref<f32, #ptr.generic_space> {
|
|
// CHECK: ptr.to_ptr
|
|
// CHECK: ptr.from_ptr
|
|
%ptr = ptr.to_ptr %mr : memref<f32, #ptr.generic_space> -> !ptr.ptr<#ptr.generic_space>
|
|
%res = ptr.from_ptr %ptr metadata %md : !ptr.ptr<#ptr.generic_space> -> memref<f32, #ptr.generic_space>
|
|
return %res : memref<f32, #ptr.generic_space>
|
|
}
|
|
|
|
// Check the folding of `to_ptr -> from_ptr` chains.
|
|
// CHECK-LABEL: @test_from_ptr_3
|
|
// CHECK-SAME: (%[[MEM_REF:.*]]: memref<f32, #ptr.generic_space>)
|
|
func.func @test_from_ptr_3(%mr0: memref<f32, #ptr.generic_space>) -> memref<f32, #ptr.generic_space> {
|
|
// CHECK-NOT: ptr.to_ptr
|
|
// CHECK-NOT: ptr.from_ptr
|
|
// CHECK: return %[[MEM_REF]]
|
|
%mda = ptr.get_metadata %mr0 : memref<f32, #ptr.generic_space>
|
|
%ptr0 = ptr.to_ptr %mr0 : memref<f32, #ptr.generic_space> -> !ptr.ptr<#ptr.generic_space>
|
|
%mrf0 = ptr.from_ptr %ptr0 metadata %mda : !ptr.ptr<#ptr.generic_space> -> memref<f32, #ptr.generic_space>
|
|
%ptr1 = ptr.to_ptr %mrf0 : memref<f32, #ptr.generic_space> -> !ptr.ptr<#ptr.generic_space>
|
|
%mrf1 = ptr.from_ptr %ptr1 metadata %mda : !ptr.ptr<#ptr.generic_space> -> memref<f32, #ptr.generic_space>
|
|
return %mrf1 : memref<f32, #ptr.generic_space>
|
|
}
|
|
|
|
/// Tests the the `to_ptr` folder.
|
|
// CHECK-LABEL: @test_to_ptr_0
|
|
// CHECK-SAME: (%[[PTR:.*]]: !ptr.ptr<#ptr.generic_space>
|
|
func.func @test_to_ptr_0(%ptr: !ptr.ptr<#ptr.generic_space>, %md: !ptr.ptr_metadata<memref<f32, #ptr.generic_space>>) -> !ptr.ptr<#ptr.generic_space> {
|
|
// CHECK: return %[[PTR]]
|
|
// CHECK-NOT: ptr.from_ptr
|
|
// CHECK-NOT: ptr.to_ptr
|
|
%mrf = ptr.from_ptr %ptr metadata %md : !ptr.ptr<#ptr.generic_space> -> memref<f32, #ptr.generic_space>
|
|
%res = ptr.to_ptr %mrf : memref<f32, #ptr.generic_space> -> !ptr.ptr<#ptr.generic_space>
|
|
return %res : !ptr.ptr<#ptr.generic_space>
|
|
}
|
|
|
|
// CHECK-LABEL: @test_to_ptr_1
|
|
// CHECK-SAME: (%[[PTR:.*]]: !ptr.ptr<#ptr.generic_space>)
|
|
func.func @test_to_ptr_1(%ptr: !ptr.ptr<#ptr.generic_space>) -> !ptr.ptr<#ptr.generic_space> {
|
|
// CHECK-NOT: ptr.from_ptr
|
|
// CHECK-NOT: ptr.to_ptr
|
|
// CHECK: return %[[PTR]]
|
|
%mrf = ptr.from_ptr %ptr : !ptr.ptr<#ptr.generic_space> -> memref<f32, #ptr.generic_space>
|
|
%res = ptr.to_ptr %mrf : memref<f32, #ptr.generic_space> -> !ptr.ptr<#ptr.generic_space>
|
|
return %res : !ptr.ptr<#ptr.generic_space>
|
|
}
|
|
|
|
// Check the folding of `from_ptr -> to_ptr` chains.
|
|
// CHECK-LABEL: @test_to_ptr_2
|
|
// CHECK-SAME: (%[[PTR:.*]]: !ptr.ptr<#ptr.generic_space>
|
|
func.func @test_to_ptr_2(%ptr0: !ptr.ptr<#ptr.generic_space>) -> !ptr.ptr<#ptr.generic_space> {
|
|
// CHECK-NOT: ptr.from_ptr
|
|
// CHECK-NOT: ptr.to_ptr
|
|
// CHECK: return %[[PTR]]
|
|
%mrf0 = ptr.from_ptr %ptr0 : !ptr.ptr<#ptr.generic_space> -> memref<f32, #ptr.generic_space>
|
|
%ptr1 = ptr.to_ptr %mrf0 : memref<f32, #ptr.generic_space> -> !ptr.ptr<#ptr.generic_space>
|
|
%mrf1 = ptr.from_ptr %ptr1 : !ptr.ptr<#ptr.generic_space> -> memref<f32, #ptr.generic_space>
|
|
%ptr2 = ptr.to_ptr %mrf1 : memref<f32, #ptr.generic_space> -> !ptr.ptr<#ptr.generic_space>
|
|
%mrf2 = ptr.from_ptr %ptr2 : !ptr.ptr<#ptr.generic_space> -> memref<f32, #ptr.generic_space>
|
|
%res = ptr.to_ptr %mrf2 : memref<f32, #ptr.generic_space> -> !ptr.ptr<#ptr.generic_space>
|
|
return %res : !ptr.ptr<#ptr.generic_space>
|
|
}
|
|
|
|
// Check the folding of chains with different metadata.
|
|
// CHECK-LABEL: @test_cast_chain_folding
|
|
// CHECK-SAME: (%[[MEM_REF:.*]]: memref<f32, #ptr.generic_space>
|
|
func.func @test_cast_chain_folding(%mr: memref<f32, #ptr.generic_space>, %md: !ptr.ptr_metadata<memref<f32, #ptr.generic_space>>) -> memref<f32, #ptr.generic_space> {
|
|
// CHECK-NOT: ptr.to_ptr
|
|
// CHECK-NOT: ptr.from_ptr
|
|
// CHECK: return %[[MEM_REF]]
|
|
%ptr1 = ptr.to_ptr %mr : memref<f32, #ptr.generic_space> -> !ptr.ptr<#ptr.generic_space>
|
|
%memrefWithOtherMd = ptr.from_ptr %ptr1 metadata %md : !ptr.ptr<#ptr.generic_space> -> memref<f32, #ptr.generic_space>
|
|
%ptr = ptr.to_ptr %memrefWithOtherMd : memref<f32, #ptr.generic_space> -> !ptr.ptr<#ptr.generic_space>
|
|
%mda = ptr.get_metadata %mr : memref<f32, #ptr.generic_space>
|
|
// The chain can be folded because: the ptr always has the same value because
|
|
// `to_ptr` is a loss-less cast and %mda comes from the original memref.
|
|
%res = ptr.from_ptr %ptr metadata %mda : !ptr.ptr<#ptr.generic_space> -> memref<f32, #ptr.generic_space>
|
|
return %res : memref<f32, #ptr.generic_space>
|
|
}
|