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
70 lines
2.8 KiB
MLIR
70 lines
2.8 KiB
MLIR
// RUN: mlir-opt %s -allow-unregistered-dialect -pass-pipeline='builtin.module(test.symbol_scope_isolated(test-one-shot-module-bufferize))' -split-input-file | FileCheck %s
|
|
|
|
"test.symbol_scope_isolated"() ({
|
|
// CHECK-LABEL: func @inner_func(
|
|
// CHECK-SAME: %[[arg0:.*]]: memref<?xf32
|
|
func.func @inner_func(%t: tensor<?xf32>) -> (tensor<?xf32>, f32) {
|
|
// CHECK-NOT: copy
|
|
%f = arith.constant 1.0 : f32
|
|
%c0 = arith.constant 0 : index
|
|
%c1 = arith.constant 1 : index
|
|
// CHECK: memref.store %{{.*}}, %[[arg0]]
|
|
%0 = tensor.insert %f into %t[%c0] : tensor<?xf32>
|
|
// CHECK: %[[load:.*]] = memref.load %[[arg0]]
|
|
%1 = tensor.extract %0[%c1] : tensor<?xf32>
|
|
// CHECK: return %[[arg0]], %[[load]] : memref<?xf32{{.*}}>, f32
|
|
return %0, %1 : tensor<?xf32>, f32
|
|
}
|
|
|
|
// CHECK-LABEL: func @call_func_with_non_tensor_return(
|
|
// CHECK-SAME: %[[arg0:.*]]: memref<?xf32
|
|
func.func @call_func_with_non_tensor_return(
|
|
%t0: tensor<?xf32> {bufferization.writable = true}) -> (f32, tensor<?xf32>) {
|
|
// CHECK-NOT: alloc
|
|
// CHECK-NOT: copy
|
|
// CHECK: %[[call:.*]]:2 = call @inner_func(%[[arg0]])
|
|
%0, %1 = call @inner_func(%t0) : (tensor<?xf32>) -> (tensor<?xf32>, f32)
|
|
// CHECK: return %[[call]]#1, %[[call]]#0 : f32, memref<?xf32{{.*}}>
|
|
return %1, %0 : f32, tensor<?xf32>
|
|
}
|
|
"test.finish" () : () -> ()
|
|
}) : () -> ()
|
|
|
|
// -----
|
|
|
|
#enc1 = #test.tensor_encoding<"hello">
|
|
#enc2 = #test.tensor_encoding<"not hello">
|
|
|
|
"test.symbol_scope_isolated"() ({
|
|
// CHECK: func @inner_func(
|
|
// CHECK-SAME: %[[arg0:.*]]: memref<?xf32, #test.memref_layout<"hello">>)
|
|
// CHECK-SAME: -> memref<?xf32, #test.memref_layout<"hello">>
|
|
func.func @inner_func(%t: tensor<?xf32, #enc1>)
|
|
-> tensor<?xf32, #enc1> {
|
|
// CHECK: return %[[arg0]]
|
|
return %t : tensor<?xf32, #enc1>
|
|
}
|
|
|
|
// CHECK: func @outer_func(
|
|
// CHECK-SAME: %[[arg0:.*]]: memref<?xf32, #test.memref_layout<"hello">>)
|
|
// CHECK-SAME: -> (memref<?xf32, #test.memref_layout<"hello">>,
|
|
// CHECK-SAME: memref<?xf32, #test.memref_layout<"not hello">>)
|
|
func.func @outer_func(%t0: tensor<?xf32, #enc1>)
|
|
-> (tensor<?xf32, #enc1>, tensor<?xf32, #enc2>) {
|
|
// CHECK: %[[call:.*]] = call @inner_func(%[[arg0]])
|
|
%0 = call @inner_func(%t0)
|
|
: (tensor<?xf32, #enc1>) -> (tensor<?xf32, #enc1>)
|
|
|
|
// CHECK: %[[local:.*]] = "test.create_memref_op"() : ()
|
|
// CHECK-SAME: -> memref<?xf32, #test.memref_layout<"not hello">>
|
|
%local = "test.create_tensor_op"() : () -> tensor<?xf32, #enc2>
|
|
// CHECK: %[[dummy:.*]] = "test.dummy_memref_op"(%[[local]])
|
|
%1 = "test.dummy_tensor_op"(%local) : (tensor<?xf32, #enc2>)
|
|
-> tensor<?xf32, #enc2>
|
|
|
|
// CHECK: return %[[call]], %[[dummy]]
|
|
return %0, %1 : tensor<?xf32, #enc1>, tensor<?xf32, #enc2>
|
|
}
|
|
"test.finish" () : () -> ()
|
|
}) : () -> ()
|