52 lines · plain
1// RUN: mlir-opt --split-input-file %s -convert-arith-to-amdgpu="allow-packed-f16-round-to-zero=true" | FileCheck %s2 3// CHECK-LABEL: @scalar_trunc4// CHECK-SAME: (%[[value:.*]]: f32)5func.func @scalar_trunc(%v: f32) -> f16{6 // CHECK: %[[poison:.*]] = llvm.mlir.poison : f327 // CHECK: %[[trunc:.*]] = rocdl.cvt.pkrtz %[[value]], %[[poison]] : vector<2xf16>8 // CHECK: %[[extract:.*]] = vector.extract %[[trunc]][0] : f16 from vector<2xf16>9 // CHECK: return %[[extract]] : f1610 %w = arith.truncf %v : f32 to f1611 return %w : f1612}13 14// CHECK-LABEL: @vector_trunc15// CHECK-SAME: (%[[value:.*]]: vector<2xf32>)16func.func @vector_trunc_short(%v: vector<2xf32>) -> vector<2xf16> {17 // CHECK: %[[elem0:.*]] = vector.extract %[[value]]18 // CHECK: %[[elem1:.*]] = vector.extract %[[value]]19 // CHECK: %[[ret:.*]] = rocdl.cvt.pkrtz %[[elem0]], %[[elem1]] : vector<2xf16>20 // CHECK: return %[[ret]]21 %w = arith.truncf %v : vector<2xf32> to vector<2xf16>22 return %w : vector<2xf16>23}24 25// CHECK-LABEL: @vector_trunc_long26// CHECK-SAME: (%[[value:.*]]: vector<9xf32>)27func.func @vector_trunc_long(%v: vector<9xf32>) -> vector<9xf16> {28 // CHECK: %[[elem0:.*]] = vector.extract %[[value]][0]29 // CHECK: %[[elem1:.*]] = vector.extract %[[value]][1]30 // CHECK: %[[packed0:.*]] = rocdl.cvt.pkrtz %[[elem0]], %[[elem1]] : vector<2xf16>31 // CHECK: %[[out0:.*]] = vector.insert_strided_slice %[[packed0]], {{.*}} {offsets = [0], strides = [1]} : vector<2xf16> into vector<9xf16>32 // CHECK: %[[elem2:.*]] = vector.extract %[[value]][2]33 // CHECK: %[[elem3:.*]] = vector.extract %[[value]][3]34 // CHECK: %[[packed1:.*]] = rocdl.cvt.pkrtz %[[elem2]], %[[elem3]] : vector<2xf16>35 // CHECK: %[[out1:.*]] = vector.insert_strided_slice %[[packed1]], %[[out0]] {offsets = [2], strides = [1]} : vector<2xf16> into vector<9xf16>36 // CHECK: %[[elem4:.*]] = vector.extract %[[value]][4]37 // CHECK: %[[elem5:.*]] = vector.extract %[[value]][5]38 // CHECK: %[[packed2:.*]] = rocdl.cvt.pkrtz %[[elem4]], %[[elem5]] : vector<2xf16>39 // CHECK: %[[out2:.*]] = vector.insert_strided_slice %[[packed2]], %[[out1]] {offsets = [4], strides = [1]} : vector<2xf16> into vector<9xf16>40 // CHECK: %[[elem6:.*]] = vector.extract %[[value]]41 // CHECK: %[[elem7:.*]] = vector.extract %[[value]]42 // CHECK: %[[packed3:.*]] = rocdl.cvt.pkrtz %[[elem6]], %[[elem7]] : vector<2xf16>43 // CHECK: %[[out3:.*]] = vector.insert_strided_slice %[[packed3]], %[[out2]] {offsets = [6], strides = [1]} : vector<2xf16> into vector<9xf16>44 // CHECK: %[[elem8:.*]] = vector.extract %[[value]]45 // CHECK: %[[packed4:.*]] = rocdl.cvt.pkrtz %[[elem8:.*]] : vector<2xf16>46 // CHECK: %[[slice:.*]] = vector.extract_strided_slice %[[packed4]] {offsets = [0], sizes = [1], strides = [1]} : vector<2xf16> to vector<1xf16>47 // CHECK: %[[out4:.*]] = vector.insert_strided_slice %[[slice]], %[[out3]] {offsets = [8], strides = [1]} : vector<1xf16> into vector<9xf16>48 // CHECK: return %[[out4]]49 %w = arith.truncf %v : vector<9xf32> to vector<9xf16>50 return %w : vector<9xf16>51}52