81 lines · plain
1// RUN: mlir-opt -split-input-file -verify-diagnostics %s2 3/// Test `to_ptr` verifiers.4func.func @invalid_to_ptr(%v: memref<f32, 0>) {5 // expected-error@+1 {{expected the input and output to have the same memory space}}6 %r = ptr.to_ptr %v : memref<f32, 0> -> !ptr.ptr<#ptr.generic_space>7 return8}9 10// -----11 12func.func @invalid_to_ptr(%v: !ptr.ptr<#ptr.generic_space>) {13 // expected-error@+1 {{the input value cannot be of type `!ptr.ptr`}}14 %r = ptr.to_ptr %v : !ptr.ptr<#ptr.generic_space> -> !ptr.ptr<#ptr.generic_space>15 return16}17 18// -----19 20func.func @invalid_load_alignment(%arg0: !ptr.ptr<#ptr.generic_space>) -> i64 {21 // expected-error@+1 {{alignment must be a power of 2}}22 %r = ptr.load %arg0 alignment = 3 : !ptr.ptr<#ptr.generic_space> -> i6423 return %r : i6424}25 26// -----27 28func.func @invalid_store_alignment(%arg0: !ptr.ptr<#ptr.generic_space>, %arg1: i64) {29 // expected-error@+1 {{alignment must be a power of 2}}30 ptr.store %arg1, %arg0 alignment = 3 : i64, !ptr.ptr<#ptr.generic_space>31 return32}33 34// -----35 36func.func @store_const(%arg0: !ptr.ptr<#test.const_memory_space>, %arg1: i64) {37 // expected-error@+1 {{memory space is read-only}}38 ptr.store %arg1, %arg0 atomic monotonic alignment = 8 : i64, !ptr.ptr<#test.const_memory_space>39 return40}41 42// -----43 44func.func @llvm_load(%arg0: !ptr.ptr<#llvm.address_space<1>>) -> (memref<f32>) {45 // expected-error@+1 {{type must be LLVM type with size, but got 'memref<f32>'}}46 %0 = ptr.load %arg0 : !ptr.ptr<#llvm.address_space<1>> -> memref<f32>47 return %0 : memref<f32>48}49 50// -----51 52func.func @llvm_store(%arg0: !ptr.ptr<#llvm.address_space<1>>, %arg1: memref<f32>) {53 // expected-error@+1 {{type must be LLVM type with size, but got 'memref<f32>'}}54 ptr.store %arg1, %arg0 : memref<f32>, !ptr.ptr<#llvm.address_space<1>>55 return56}57 58// -----59 60func.func @ptr_add_mismatch(%ptrs: tensor<8x!ptr.ptr<#ptr.generic_space>>, %offsets: vector<8xi64>) -> tensor<8x!ptr.ptr<#ptr.generic_space>> {61 // expected-error@+1 {{the shaped containers type must match}}62 %res = ptr.ptr_add %ptrs, %offsets : tensor<8x!ptr.ptr<#ptr.generic_space>>, vector<8xi64>63 return %res : tensor<8x!ptr.ptr<#ptr.generic_space>>64}65 66// -----67 68func.func @ptr_add_shape_mismatch(%ptrs: tensor<8x!ptr.ptr<#ptr.generic_space>>, %offsets: tensor<4xi64>) -> tensor<8x!ptr.ptr<#ptr.generic_space>> {69 // expected-error@+1 {{shapes of base and offset must match}}70 %res = ptr.ptr_add %ptrs, %offsets : tensor<8x!ptr.ptr<#ptr.generic_space>>, tensor<4xi64>71 return %res : tensor<8x!ptr.ptr<#ptr.generic_space>>72}73 74// -----75 76func.func @ptr_diff_mismatch(%lhs: tensor<8x!ptr.ptr<#ptr.generic_space>>, %rhs: tensor<8x!ptr.ptr<#ptr.generic_space>>) -> vector<8xi64> {77 // expected-error@+1 {{the result to have the same container type as the operands when operands are shaped}}78 %res = ptr.ptr_diff %lhs, %rhs : tensor<8x!ptr.ptr<#ptr.generic_space>> -> vector<8xi64>79 return %res : vector<8xi64>80}81