1*3cdac067SFinn Plummer; RUN: llc -O0 -mtriple=spirv1.5-unknown-unknown %s -o - | FileCheck %s --check-prefixes=CHECK,CHECK-EXP 2*3cdac067SFinn Plummer; RUN: llc -O0 -mtriple=spirv1.6-unknown-unknown %s -o - | FileCheck %s --check-prefixes=CHECK,CHECK-DOT 3*3cdac067SFinn Plummer; RUN: llc -O0 -mtriple=spirv-unknown-unknown -spirv-ext=+SPV_KHR_integer_dot_product %s -o - | FileCheck %s --check-prefixes=CHECK,CHECK-DOT,CHECK-EXT 4*3cdac067SFinn Plummer; RUN: %if spirv-tools %{ llc -verify-machineinstrs -O0 -mtriple=spirv1.5-unknown-unknown %s -o - -filetype=obj | spirv-val %} 5*3cdac067SFinn Plummer; RUN: %if spirv-tools %{ llc -verify-machineinstrs -O0 -mtriple=spirv1.6-unknown-unknown %s -o - -filetype=obj | spirv-val %} 6*3cdac067SFinn Plummer; RUN: %if spirv-tools %{ llc -verify-machineinstrs -O0 -mtriple=spirv-unknown-unknown -spirv-ext=+SPV_KHR_integer_dot_product %s -o - -filetype=obj | spirv-val %} 7*3cdac067SFinn Plummer 8*3cdac067SFinn Plummer; CHECK-DOT: OpCapability DotProduct 9*3cdac067SFinn Plummer; CHECK-DOT: OpCapability DotProductInput4x8BitPacked 10*3cdac067SFinn Plummer; CHECK-EXT: OpExtension "SPV_KHR_integer_dot_product" 11*3cdac067SFinn Plummer 12*3cdac067SFinn Plummer; CHECK: %[[#int_32:]] = OpTypeInt 32 0 13*3cdac067SFinn Plummer; CHECK-EXP-DAG: %[[#int_8:]] = OpTypeInt 8 0 14*3cdac067SFinn Plummer; CHECK-EXP-DAG: %[[#zero:]] = OpConstantNull %[[#int_8]] 15*3cdac067SFinn Plummer; CHECK-EXP-DAG: %[[#eight:]] = OpConstant %[[#int_8]] 8 16*3cdac067SFinn Plummer; CHECK-EXP-DAG: %[[#sixteen:]] = OpConstant %[[#int_8]] 16 17*3cdac067SFinn Plummer; CHECK-EXP-DAG: %[[#twentyfour:]] = OpConstant %[[#int_8]] 24 18*3cdac067SFinn Plummer 19*3cdac067SFinn Plummer; CHECK-LABEL: Begin function test_dot 20*3cdac067SFinn Plummerdefine noundef i32 @test_dot(i32 noundef %a, i32 noundef %b, i32 noundef %c) { 21*3cdac067SFinn Plummerentry: 22*3cdac067SFinn Plummer; CHECK: %[[#A:]] = OpFunctionParameter %[[#int_32]] 23*3cdac067SFinn Plummer; CHECK: %[[#B:]] = OpFunctionParameter %[[#int_32]] 24*3cdac067SFinn Plummer; CHECK: %[[#C:]] = OpFunctionParameter %[[#int_32]] 25*3cdac067SFinn Plummer 26*3cdac067SFinn Plummer; Test that we use the dot product op when capabilities allow 27*3cdac067SFinn Plummer 28*3cdac067SFinn Plummer; CHECK-DOT: %[[#DOT:]] = OpSDot %[[#int_32]] %[[#A]] %[[#B]] 29*3cdac067SFinn Plummer; CHECK-DOT: %[[#RES:]] = OpIAdd %[[#int_32]] %[[#DOT]] %[[#C]] 30*3cdac067SFinn Plummer 31*3cdac067SFinn Plummer; Test expansion is used when spirv dot product capabilities aren't available: 32*3cdac067SFinn Plummer 33*3cdac067SFinn Plummer; First element of the packed vector 34*3cdac067SFinn Plummer; CHECK-EXP: %[[#A0:]] = OpBitFieldSExtract %[[#int_32]] %[[#A]] %[[#zero]] %[[#eight]] 35*3cdac067SFinn Plummer; CHECK-EXP: %[[#B0:]] = OpBitFieldSExtract %[[#int_32]] %[[#B]] %[[#zero]] %[[#eight]] 36*3cdac067SFinn Plummer; CHECK-EXP: %[[#MUL0:]] = OpIMul %[[#int_32]] %[[#A0]] %[[#B0]] 37*3cdac067SFinn Plummer; CHECK-EXP: %[[#MASK0:]] = OpBitFieldSExtract %[[#int_32]] %[[#MUL0]] %[[#zero]] %[[#eight]] 38*3cdac067SFinn Plummer; CHECK-EXP: %[[#ACC0:]] = OpIAdd %[[#int_32]] %[[#C]] %[[#MASK0]] 39*3cdac067SFinn Plummer 40*3cdac067SFinn Plummer; Second element of the packed vector 41*3cdac067SFinn Plummer; CHECK-EXP: %[[#A1:]] = OpBitFieldSExtract %[[#int_32]] %[[#A]] %[[#eight]] %[[#eight]] 42*3cdac067SFinn Plummer; CHECK-EXP: %[[#B1:]] = OpBitFieldSExtract %[[#int_32]] %[[#B]] %[[#eight]] %[[#eight]] 43*3cdac067SFinn Plummer; CHECK-EXP: %[[#MUL1:]] = OpIMul %[[#int_32]] %[[#A1]] %[[#B1]] 44*3cdac067SFinn Plummer; CHECK-EXP: %[[#MASK1:]] = OpBitFieldSExtract %[[#int_32]] %[[#MUL1]] %[[#zero]] %[[#eight]] 45*3cdac067SFinn Plummer; CHECK-EXP: %[[#ACC1:]] = OpIAdd %[[#int_32]] %[[#ACC0]] %[[#MASK1]] 46*3cdac067SFinn Plummer 47*3cdac067SFinn Plummer; Third element of the packed vector 48*3cdac067SFinn Plummer; CHECK-EXP: %[[#A2:]] = OpBitFieldSExtract %[[#int_32]] %[[#A]] %[[#sixteen]] %[[#eight]] 49*3cdac067SFinn Plummer; CHECK-EXP: %[[#B2:]] = OpBitFieldSExtract %[[#int_32]] %[[#B]] %[[#sixteen]] %[[#eight]] 50*3cdac067SFinn Plummer; CHECK-EXP: %[[#MUL2:]] = OpIMul %[[#int_32]] %[[#A2]] %[[#B2]] 51*3cdac067SFinn Plummer; CHECK-EXP: %[[#MASK2:]] = OpBitFieldSExtract %[[#int_32]] %[[#MUL2]] %[[#zero]] %[[#eight]] 52*3cdac067SFinn Plummer; CHECK-EXP: %[[#ACC2:]] = OpIAdd %[[#int_32]] %[[#ACC1]] %[[#MASK2]] 53*3cdac067SFinn Plummer 54*3cdac067SFinn Plummer; Fourth element of the packed vector 55*3cdac067SFinn Plummer; CHECK-EXP: %[[#A3:]] = OpBitFieldSExtract %[[#int_32]] %[[#A]] %[[#twentyfour]] %[[#eight]] 56*3cdac067SFinn Plummer; CHECK-EXP: %[[#B3:]] = OpBitFieldSExtract %[[#int_32]] %[[#B]] %[[#twentyfour]] %[[#eight]] 57*3cdac067SFinn Plummer; CHECK-EXP: %[[#MUL3:]] = OpIMul %[[#int_32]] %[[#A3]] %[[#B3]] 58*3cdac067SFinn Plummer; CHECK-EXP: %[[#MASK3:]] = OpBitFieldSExtract %[[#int_32]] %[[#MUL3]] %[[#zero]] %[[#eight]] 59*3cdac067SFinn Plummer 60*3cdac067SFinn Plummer; CHECK-EXP: %[[#RES:]] = OpIAdd %[[#int_32]] %[[#ACC2]] %[[#MASK3]] 61*3cdac067SFinn Plummer; CHECK: OpReturnValue %[[#RES]] 62*3cdac067SFinn Plummer %spv.dot = call i32 @llvm.spv.dot4add.i8packed(i32 %a, i32 %b, i32 %c) 63*3cdac067SFinn Plummer 64*3cdac067SFinn Plummer ret i32 %spv.dot 65*3cdac067SFinn Plummer} 66