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
57 lines
1.9 KiB
MLIR
57 lines
1.9 KiB
MLIR
// RUN: mlir-opt -allow-unregistered-dialect -split-input-file -remove-shape-constraints -canonicalize <%s | FileCheck %s --dump-input=fail --check-prefixes=CANON,CHECK-BOTH
|
|
// RUN: mlir-opt -allow-unregistered-dialect -split-input-file -remove-shape-constraints <%s | FileCheck %s --dump-input=fail --check-prefixes=REPLACE,CHECK-BOTH
|
|
|
|
// -----
|
|
// Check that cstr_broadcastable is removed.
|
|
//
|
|
// CHECK-BOTH: func @f
|
|
func.func @f(%arg0 : !shape.shape, %arg1 : !shape.shape) -> index {
|
|
// REPLACE-NEXT: %[[WITNESS:.+]] = shape.const_witness true
|
|
// REPLACE-NOT: shape.cstr_eq
|
|
// REPLACE: shape.assuming %[[WITNESS]]
|
|
// CANON-NEXT: test.source
|
|
// CANON-NEXT: return
|
|
%0 = shape.cstr_broadcastable %arg0, %arg1 : !shape.shape, !shape.shape
|
|
%1 = shape.assuming %0 -> index {
|
|
%2 = "test.source"() : () -> (index)
|
|
shape.assuming_yield %2 : index
|
|
}
|
|
return %1 : index
|
|
}
|
|
|
|
// -----
|
|
// Check that cstr_eq is removed.
|
|
//
|
|
// CHECK-BOTH: func @f
|
|
func.func @f(%arg0 : !shape.shape, %arg1 : !shape.shape) -> index {
|
|
// REPLACE-NEXT: %[[WITNESS:.+]] = shape.const_witness true
|
|
// REPLACE-NOT: shape.cstr_eq
|
|
// REPLACE: shape.assuming %[[WITNESS]]
|
|
// CANON-NEXT: test.source
|
|
// CANON-NEXT: return
|
|
%0 = shape.cstr_eq %arg0, %arg1 : !shape.shape, !shape.shape
|
|
%1 = shape.assuming %0 -> index {
|
|
%2 = "test.source"() : () -> (index)
|
|
shape.assuming_yield %2 : index
|
|
}
|
|
return %1 : index
|
|
}
|
|
|
|
// -----
|
|
// With a non-const value, we cannot fold away the code, but all constraints
|
|
// should be removed still.
|
|
//
|
|
// CHECK-BOTH: func @f
|
|
func.func @f(%arg0 : !shape.shape, %arg1 : !shape.shape) -> index {
|
|
// CANON-NEXT: test.source
|
|
// CANON-NEXT: return
|
|
%0 = shape.cstr_broadcastable %arg0, %arg1 : !shape.shape, !shape.shape
|
|
%1 = shape.cstr_eq %arg0, %arg1 : !shape.shape, !shape.shape
|
|
%2 = shape.assuming_all %0, %1
|
|
%3 = shape.assuming %0 -> index {
|
|
%4 = "test.source"() : () -> (index)
|
|
shape.assuming_yield %4 : index
|
|
}
|
|
return %3 : index
|
|
}
|