You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
22 lines
916 B
22 lines
916 B
4 months ago
|
// RUN: mlir-opt -verify-diagnostics %s | mlir-opt | FileCheck %s
|
||
|
|
|
||
|
|
func @avx512_mask_rndscale(%a: vector<16xf32>, %b: vector<8xf64>, %i32: i32, %i16: i16, %i8: i8)
|
||
|
|
-> (vector<16xf32>, vector<8xf64>)
|
||
|
|
{
|
||
|
|
// CHECK: avx512.mask.rndscale {{.*}}: vector<16xf32>
|
||
|
|
%0 = avx512.mask.rndscale %a, %i32, %a, %i16, %i32 : vector<16xf32>
|
||
|
|
// CHECK: avx512.mask.rndscale {{.*}}: vector<8xf64>
|
||
|
|
%1 = avx512.mask.rndscale %b, %i32, %b, %i8, %i32 : vector<8xf64>
|
||
|
|
return %0, %1: vector<16xf32>, vector<8xf64>
|
||
|
|
}
|
||
|
|
|
||
|
|
func @avx512_scalef(%a: vector<16xf32>, %b: vector<8xf64>, %i32: i32, %i16: i16, %i8: i8)
|
||
|
|
-> (vector<16xf32>, vector<8xf64>)
|
||
|
|
{
|
||
|
|
// CHECK: avx512.mask.scalef {{.*}}: vector<16xf32>
|
||
|
|
%0 = avx512.mask.scalef %a, %a, %a, %i16, %i32: vector<16xf32>
|
||
|
|
// CHECK: avx512.mask.scalef {{.*}}: vector<8xf64>
|
||
|
|
%1 = avx512.mask.scalef %b, %b, %b, %i8, %i32 : vector<8xf64>
|
||
|
|
return %0, %1: vector<16xf32>, vector<8xf64>
|
||
|
|
}
|