22 lines · plain
1; RUN: llc -verify-machineinstrs -O0 -mtriple=spirv32-unknown-unknown --spirv-ext=+SPV_KHR_bfloat16 %s -o - | FileCheck %s2; RUN: %if spirv-tools %{ llc -O0 -mtriple=spirv64-unknown-unknown --spirv-ext=+SPV_KHR_bfloat16 %s -o - -filetype=obj | spirv-val %}3 4; CHECK-DAG: OpCapability BFloat16TypeKHR5; CHECK-DAG: OpCapability BFloat16DotProductKHR6; CHECK-DAG: OpExtension "SPV_KHR_bfloat16"7; CHECK: %[[#BFLOAT:]] = OpTypeFloat 16 08; CHECK: %[[#]] = OpTypeVector %[[#BFLOAT]] 29; CHECK: OpDot10 11declare spir_func bfloat @_Z3dotDv2_u6__bf16Dv2_S_(<2 x bfloat>, <2 x bfloat>)12 13define spir_kernel void @test() {14entry:15 %addrA = alloca <2 x bfloat>16 %addrB = alloca <2 x bfloat>17 %dataA = load <2 x bfloat>, ptr %addrA18 %dataB = load <2 x bfloat>, ptr %addrB19 %call = call spir_func bfloat @_Z3dotDv2_u6__bf16Dv2_S_(<2 x bfloat> %dataA, <2 x bfloat> %dataB)20 ret void21}22