1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
|
;=========================== begin_copyright_notice ============================
;
; Copyright (C) 2024 Intel Corporation
;
; SPDX-License-Identifier: MIT
;
;============================ end_copyright_notice =============================
; RUN: igc_opt -igc-spv-subgroup-mma-resolution -S --platformpvc %s 2>&1 | FileCheck %s
; ------------------------------------------------
; SpvSubgroupMMAResolution - basic test
; ------------------------------------------------
target triple = "spir64-unknown-unknown"
declare spir_func signext i16 @_Z45__spirv_SubgroupMatrixMultiplyAccumulateINTELisDv8_isi(i32, i16 signext, <8 x i32>, i16 signext, i32)
declare spir_func half @_Z45__spirv_SubgroupMatrixMultiplyAccumulateINTELisDv8_f16(i32, i16 signext, <8 x i32>, half, i32)
declare spir_func signext i16 @_Z45__spirv_SubgroupMatrixMultiplyAccumulateINTELisDv8_wrong_type(i32, i16 signext, i32 addrspace(1)*, i16 signext, i32)
declare spir_func signext i16 @_Z45__spirv_SubgroupMatrixMultiplyAccumulateINTELisDv8_wrong_K_type(i16, i16 signext, <8 x i32>, i16 signext, i32)
define spir_kernel void @test(<8 x i32> %b, i32 addrspace(1)* %b1) {
entry:
;CHECK: call i16 @__builtin_IB_sub_group16_fdpas_bf_bf_bf_bf_8_1(i16 0, i16 0, <8 x i32> %b)
%call0 = call spir_func i16 @_Z45__spirv_SubgroupMatrixMultiplyAccumulateINTELisDv8_isi(i32 16, i16 0, <8 x i32> %b, i16 0, i32 12300)
;CHECK: error: __spirv_SubgroupMatrixMultiplyAccumulateINTEL: expected Operands to be one of these combinations:
;CHECK-NEXT: 12300: MatrixCBFloat16INTEL | MatrixResultBFloat16INTEL | MatrixAPackedBFloat16INTEL | MatrixBPackedBFloat16INTEL
;CHECK-NEXT: for K Dim = 16, for Result element type int16_t, for A element type int16_t, for B element type int32_t, for targeted HW.
;CHECK-NEXT: Actual: 12301: MatrixASignedComponentsINTEL | MatrixCBFloat16INTEL | MatrixResultBFloat16INTEL | MatrixAPackedBFloat16INTEL | MatrixBPackedBFloat16INTEL
%call1 = call spir_func i16 @_Z45__spirv_SubgroupMatrixMultiplyAccumulateINTELisDv8_isi(i32 16, i16 0, <8 x i32> %b, i16 0, i32 12301)
;CHECK: for K Dim = 16, for Result element type float16_t, for A element type int16_t, for B element type int32_t, for targeted HW.
%call2 = call spir_func half @_Z45__spirv_SubgroupMatrixMultiplyAccumulateINTELisDv8_f16(i32 16, i16 0, <8 x i32> %b, half 0.0, i32 12300)
;CHECK: error: __spirv_SubgroupMatrixMultiplyAccumulateINTEL: expected Matrix B to be a scalar or vector of int32_t, int16_t, float32_t, or float16_t for targeted HW
%call3 = call spir_func i16 @_Z45__spirv_SubgroupMatrixMultiplyAccumulateINTELisDv8_wrong_type(i32 16, i16 0, i32 addrspace(1)* %b1, i16 0, i32 12300)
;CHECK: error: __spirv_SubgroupMatrixMultiplyAccumulateINTEL: K Dim argument must be a constant scalar 32-bit integer
%call4 = call spir_func i16 @_Z45__spirv_SubgroupMatrixMultiplyAccumulateINTELisDv8_wrong_K_type(i16 16, i16 0, <8 x i32> %b, i16 0, i32 12300)
ret void
}
!igc.functions = !{!0}
!0 = !{void (<8 x i32>, i32 addrspace(1)*)* @test, !1}
!1 = !{!2, !3}
!2 = !{!"function_type", i32 0}
!3 = !{!"sub_group_size", i32 16}
|
