forked from OSchip/llvm-project
222 lines
5.5 KiB
MLIR
222 lines
5.5 KiB
MLIR
// RUN: mlir-opt -split-input-file -verify-diagnostics %s | FileCheck %s
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// spv.FAdd
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func @fadd_scalar(%arg: f32) -> f32 {
|
|
// CHECK: spv.FAdd
|
|
%0 = spv.FAdd %arg, %arg : f32
|
|
return %0 : f32
|
|
}
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// spv.FDiv
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func @fdiv_scalar(%arg: f32) -> f32 {
|
|
// CHECK: spv.FDiv
|
|
%0 = spv.FDiv %arg, %arg : f32
|
|
return %0 : f32
|
|
}
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// spv.FMod
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func @fmod_scalar(%arg: f32) -> f32 {
|
|
// CHECK: spv.FMod
|
|
%0 = spv.FMod %arg, %arg : f32
|
|
return %0 : f32
|
|
}
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// spv.FMul
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func @fmul_scalar(%arg: f32) -> f32 {
|
|
// CHECK: spv.FMul
|
|
%0 = spv.FMul %arg, %arg : f32
|
|
return %0 : f32
|
|
}
|
|
|
|
func @fmul_vector(%arg: vector<4xf32>) -> vector<4xf32> {
|
|
// CHECK: spv.FMul
|
|
%0 = spv.FMul %arg, %arg : vector<4xf32>
|
|
return %0 : vector<4xf32>
|
|
}
|
|
|
|
// -----
|
|
|
|
func @fmul_i32(%arg: i32) -> i32 {
|
|
// expected-error @+1 {{operand #0 must be 16/32/64-bit float or vector of 16/32/64-bit float values}}
|
|
%0 = spv.FMul %arg, %arg : i32
|
|
return %0 : i32
|
|
}
|
|
|
|
// -----
|
|
|
|
func @fmul_bf16(%arg: bf16) -> bf16 {
|
|
// expected-error @+1 {{operand #0 must be 16/32/64-bit float or vector of 16/32/64-bit float values}}
|
|
%0 = spv.FMul %arg, %arg : bf16
|
|
return %0 : bf16
|
|
}
|
|
|
|
// -----
|
|
|
|
func @fmul_tensor(%arg: tensor<4xf32>) -> tensor<4xf32> {
|
|
// expected-error @+1 {{operand #0 must be 16/32/64-bit float or vector of 16/32/64-bit float values}}
|
|
%0 = spv.FMul %arg, %arg : tensor<4xf32>
|
|
return %0 : tensor<4xf32>
|
|
}
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// spv.FNegate
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func @fnegate_scalar(%arg: f32) -> f32 {
|
|
// CHECK: spv.FNegate
|
|
%0 = spv.FNegate %arg : f32
|
|
return %0 : f32
|
|
}
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// spv.FRem
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func @frem_scalar(%arg: f32) -> f32 {
|
|
// CHECK: spv.FRem
|
|
%0 = spv.FRem %arg, %arg : f32
|
|
return %0 : f32
|
|
}
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// spv.FSub
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func @fsub_scalar(%arg: f32) -> f32 {
|
|
// CHECK: spv.FSub
|
|
%0 = spv.FSub %arg, %arg : f32
|
|
return %0 : f32
|
|
}
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// spv.IAdd
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func @iadd_scalar(%arg: i32) -> i32 {
|
|
// CHECK: spv.IAdd
|
|
%0 = spv.IAdd %arg, %arg : i32
|
|
return %0 : i32
|
|
}
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// spv.IMul
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func @imul_scalar(%arg: i32) -> i32 {
|
|
// CHECK: spv.IMul
|
|
%0 = spv.IMul %arg, %arg : i32
|
|
return %0 : i32
|
|
}
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// spv.ISub
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func @isub_scalar(%arg: i32) -> i32 {
|
|
// CHECK: spv.ISub
|
|
%0 = spv.ISub %arg, %arg : i32
|
|
return %0 : i32
|
|
}
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// spv.SDiv
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func @sdiv_scalar(%arg: i32) -> i32 {
|
|
// CHECK: spv.SDiv
|
|
%0 = spv.SDiv %arg, %arg : i32
|
|
return %0 : i32
|
|
}
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// spv.SMod
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func @smod_scalar(%arg: i32) -> i32 {
|
|
// CHECK: spv.SMod
|
|
%0 = spv.SMod %arg, %arg : i32
|
|
return %0 : i32
|
|
}
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// spv.SNegate
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func @snegate_scalar(%arg: i32) -> i32 {
|
|
// CHECK: spv.SNegate
|
|
%0 = spv.SNegate %arg : i32
|
|
return %0 : i32
|
|
}
|
|
|
|
// -----
|
|
//===----------------------------------------------------------------------===//
|
|
// spv.SRem
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func @srem_scalar(%arg: i32) -> i32 {
|
|
// CHECK: spv.SRem
|
|
%0 = spv.SRem %arg, %arg : i32
|
|
return %0 : i32
|
|
}
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// spv.UDiv
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func @udiv_scalar(%arg: i32) -> i32 {
|
|
// CHECK: spv.UDiv
|
|
%0 = spv.UDiv %arg, %arg : i32
|
|
return %0 : i32
|
|
}
|
|
|
|
// -----
|
|
|
|
//===----------------------------------------------------------------------===//
|
|
// spv.UMod
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
func @umod_scalar(%arg: i32) -> i32 {
|
|
// CHECK: spv.UMod
|
|
%0 = spv.UMod %arg, %arg : i32
|
|
return %0 : i32
|
|
}
|
|
|