Please use GitHub pull requests for new patches. Avoid migrating existing patches. Phabricator shutdown timeline
Changeset View
Changeset View
Standalone View
Standalone View
mlir/test/Conversion/MemRefToLLVM/generic-functions.mlir
// RUN: mlir-opt -pass-pipeline="builtin.module(finalize-memref-to-llvm{use-generic-functions=1})" -split-input-file %s \ | // RUN: mlir-opt -pass-pipeline="builtin.module(finalize-memref-to-llvm{use-generic-functions=1 use-opaque-pointers=1})" -split-input-file %s \ | ||||
// RUN: | FileCheck %s --check-prefix="CHECK-NOTALIGNED" | // RUN: | FileCheck %s --check-prefix="CHECK-NOTALIGNED" | ||||
// RUN: mlir-opt -pass-pipeline="builtin.module(finalize-memref-to-llvm{use-generic-functions=1 use-aligned-alloc=1})" -split-input-file %s \ | // RUN: mlir-opt -pass-pipeline="builtin.module(finalize-memref-to-llvm{use-generic-functions=1 use-aligned-alloc=1 use-opaque-pointers=1})" -split-input-file %s \ | ||||
// RUN: | FileCheck %s --check-prefix="CHECK-ALIGNED" | // RUN: | FileCheck %s --check-prefix="CHECK-ALIGNED" | ||||
// CHECK-LABEL: func @alloc() | // CHECK-LABEL: func @alloc() | ||||
func.func @zero_d_alloc() -> memref<f32> { | func.func @zero_d_alloc() -> memref<f32> { | ||||
// CHECK-NOTALIGNED: llvm.call @_mlir_memref_to_llvm_alloc(%{{.*}}) : (i64) -> !llvm.ptr<i8> | // CHECK-NOTALIGNED: llvm.call @_mlir_memref_to_llvm_alloc(%{{.*}}) : (i64) -> !llvm.ptr | ||||
// CHECK-ALIGNED: llvm.call @_mlir_memref_to_llvm_aligned_alloc(%{{.*}}, %{{.*}}) : (i64, i64) -> !llvm.ptr<i8> | // CHECK-ALIGNED: llvm.call @_mlir_memref_to_llvm_aligned_alloc(%{{.*}}, %{{.*}}) : (i64, i64) -> !llvm.ptr | ||||
%0 = memref.alloc() : memref<f32> | %0 = memref.alloc() : memref<f32> | ||||
return %0 : memref<f32> | return %0 : memref<f32> | ||||
} | } | ||||
// ----- | // ----- | ||||
// CHECK-LABEL: func @dealloc() | // CHECK-LABEL: func @dealloc() | ||||
func.func @dealloc(%arg0: memref<f32>) { | func.func @dealloc(%arg0: memref<f32>) { | ||||
// CHECK-NOTALIGNED: llvm.call @_mlir_memref_to_llvm_free(%{{.*}}) : (!llvm.ptr<i8>) -> () | // CHECK-NOTALIGNED: llvm.call @_mlir_memref_to_llvm_free(%{{.*}}) : (!llvm.ptr) -> () | ||||
// CHECK-ALIGNED: llvm.call @_mlir_memref_to_llvm_free(%{{.*}}) : (!llvm.ptr<i8>) -> () | // CHECK-ALIGNED: llvm.call @_mlir_memref_to_llvm_free(%{{.*}}) : (!llvm.ptr) -> () | ||||
memref.dealloc %arg0 : memref<f32> | memref.dealloc %arg0 : memref<f32> | ||||
return | return | ||||
} | } |