// RUN: %clang_cc1 -no-enable-noundef-analysis -fenable-matrix -triple x86_64-apple-darwin %s -emit-llvm -disable-llvm-passes -o - | FileCheck --check-prefixes=COMMON,CHECK64 %s // RUN: %clang_cc1 -no-enable-noundef-analysis -fenable-matrix -triple i386-apple-darwin %s -emit-llvm -disable-llvm-passes -o - | FileCheck --check-prefixes=COMMON,CHECK32 %s // Also check we do not crash when running some middle-end passes. Most // importantly this includes the IR verifier, to ensure we emit valid IR. // RUN: %clang_cc1 -fenable-matrix -emit-llvm -triple x86_64-apple-darwin %s -o %t // Tests for the matrix type builtins. typedef double dx5x5_t __attribute__((matrix_type(5, 5))); typedef float fx2x3_t __attribute__((matrix_type(2, 3))); typedef float fx3x2_t __attribute__((matrix_type(3, 2))); typedef int ix20x4_t __attribute__((matrix_type(20, 4))); typedef int ix4x20_t __attribute__((matrix_type(4, 20))); typedef unsigned ux1x6_t __attribute__((matrix_type(1, 6))); typedef unsigned ux6x1_t __attribute__((matrix_type(6, 1))); void transpose_double_5x5(dx5x5_t *a) { // COMMON-LABEL: define{{.*}} void @transpose_double_5x5( // CHECK32: [[A:%.*]] = load <25 x double>, ptr {{.*}}, align 4 // CHECK64: [[A:%.*]] = load <25 x double>, ptr {{.*}}, align 8 // COMMON-NEXT: [[TRANS:%.*]] = call <25 x double> @llvm.matrix.transpose.v25f64(<25 x double> [[A]], i32 5, i32 5) // CHECK32-NEXT: store <25 x double> [[TRANS]], ptr %a_t, align 4 // CHECK64-NEXT: store <25 x double> [[TRANS]], ptr %a_t, align 8 dx5x5_t a_t = __builtin_matrix_transpose(*a); } void transpose_float_3x2(fx3x2_t *a) { // COMMON-LABEL: define{{.*}} void @transpose_float_3x2( // COMMON: [[A:%.*]] = load <6 x float>, ptr {{.*}}, align 4 // COMMON-NEXT: [[TRANS:%.*]] = call <6 x float> @llvm.matrix.transpose.v6f32(<6 x float> [[A]], i32 3, i32 2) // COMMON-NEXT: store <6 x float> [[TRANS]], ptr %a_t, align 4 fx2x3_t a_t = __builtin_matrix_transpose(*a); } void transpose_int_20x4(ix20x4_t *a) { // COMMON-LABEL: define{{.*}} void @transpose_int_20x4( // COMMON: [[A:%.*]] = load <80 x i32>, ptr {{.*}}, align 4 // COMMON-NEXT: [[TRANS:%.*]] = call <80 x i32> @llvm.matrix.transpose.v80i32(<80 x i32> [[A]], i32 20, i32 4) // COMMON-NEXT: store <80 x i32> [[TRANS]], ptr %a_t, align 4 ix4x20_t a_t = __builtin_matrix_transpose(*a); } struct Foo { ux1x6_t in; ux6x1_t out; }; void transpose_struct_member(struct Foo *F) { // COMMON-LABEL: define{{.*}} void @transpose_struct_member( // COMMON: [[M:%.*]] = load <6 x i32>, ptr {{.*}}, align 4 // COMMON-NEXT: [[M_T:%.*]] = call <6 x i32> @llvm.matrix.transpose.v6i32(<6 x i32> [[M]], i32 1, i32 6) // CHECK32-NEXT: [[F_ADDR:%.*]] = load ptr, ptr %F.addr, align 4 // CHECK64-NEXT: [[F_ADDR:%.*]] = load ptr, ptr %F.addr, align 8 // COMMON-NEXT: [[OUT_PTR:%.*]] = getelementptr inbounds nuw %struct.Foo, ptr [[F_ADDR]], i32 0, i32 1 // COMMON-NEXT: store <6 x i32> [[M_T]], ptr [[OUT_PTR]], align 4 F->out = __builtin_matrix_transpose(F->in); } void transpose_transpose_struct_member(struct Foo *F) { // COMMON-LABEL: define{{.*}} void @transpose_transpose_struct_member( // COMMON: [[M:%.*]] = load <6 x i32>, ptr {{.*}}, align 4 // COMMON-NEXT: [[M_T:%.*]] = call <6 x i32> @llvm.matrix.transpose.v6i32(<6 x i32> [[M]], i32 1, i32 6) // COMMON-NEXT: [[M_T2:%.*]] = call <6 x i32> @llvm.matrix.transpose.v6i32(<6 x i32> [[M_T]], i32 6, i32 1) // CHECK32-NEXT: [[F_ADDR:%.*]] = load ptr, ptr %F.addr, align 4 // CHECK64-NEXT: [[F_ADDR:%.*]] = load ptr, ptr %F.addr, align 8 // COMMON-NEXT: [[IN_PTR:%.*]] = getelementptr inbounds nuw %struct.Foo, ptr [[F_ADDR]], i32 0, i32 0 // COMMON-NEXT: store <6 x i32> [[M_T2]], ptr [[IN_PTR]], align 4 F->in = __builtin_matrix_transpose(__builtin_matrix_transpose(F->in)); } dx5x5_t get_matrix(void); void transpose_rvalue(void) { // COMMON-LABEL: define{{.*}} void @transpose_rvalue() // COMMON-NEXT: entry: // CHECK32-NEXT: [[M_T_ADDR:%.*]] = alloca [25 x double], align 4 // CHECK64-NEXT: [[M_T_ADDR:%.*]] = alloca [25 x double], align 8 // COMMON-NEXT: [[CALL:%.*]] = call <25 x double> @get_matrix() // COMMON-NEXT: [[M_T:%.*]] = call <25 x double> @llvm.matrix.transpose.v25f64(<25 x double> [[CALL]], i32 5, i32 5) // CHECK32-NEXT: store <25 x double> [[M_T]], ptr [[M_T_ADDR]], align 4 // CHECK64-NEXT: store <25 x double> [[M_T]], ptr [[M_T_ADDR]], align 8 dx5x5_t m_t = __builtin_matrix_transpose(get_matrix()); } const dx5x5_t global_matrix; void transpose_global(void) { // COMMON-LABEL: define{{.*}} void @transpose_global() // COMMON-NEXT: entry: // CHECK32-NEXT: [[M_T_ADDR:%.*]] = alloca [25 x double], align 4 // CHECK32-NEXT: [[GLOBAL_MATRIX:%.*]] = load <25 x double>, ptr @global_matrix, align 4 // CHECK64-NEXT: [[M_T_ADDR:%.*]] = alloca [25 x double], align 8 // CHECK64-NEXT: [[GLOBAL_MATRIX:%.*]] = load <25 x double>, ptr @global_matrix, align 8 // COMMON-NEXT: [[M_T:%.*]] = call <25 x double> @llvm.matrix.transpose.v25f64(<25 x double> [[GLOBAL_MATRIX]], i32 5, i32 5) // CHECK32-NEXT: store <25 x double> [[M_T]], ptr [[M_T_ADDR]], align 4 // CHECK64-NEXT: store <25 x double> [[M_T]], ptr [[M_T_ADDR]], align 8 dx5x5_t m_t = __builtin_matrix_transpose(global_matrix); } void column_major_load_with_const_stride_double(double *Ptr) { // COMMON-LABEL: define{{.*}} void @column_major_load_with_const_stride_double(ptr %Ptr) // CHECK32: [[PTR:%.*]] = load ptr, ptr %Ptr.addr, align 4 // CHECK32-NEXT: call <25 x double> @llvm.matrix.column.major.load.v25f64.i32(ptr align 4 [[PTR]], i32 5, i1 false, i32 5, i32 5) // CHECK64: [[PTR:%.*]] = load ptr, ptr %Ptr.addr, align 8 // CHECK64-NEXT: call <25 x double> @llvm.matrix.column.major.load.v25f64.i64(ptr align 8 [[PTR]], i64 5, i1 false, i32 5, i32 5) dx5x5_t m_a1 = __builtin_matrix_column_major_load(Ptr, 5, 5, 5); } void column_major_load_with_const_stride2_double(double *Ptr) { // COMMON-LABEL: define{{.*}} void @column_major_load_with_const_stride2_double(ptr %Ptr) // CHECK32: [[PTR:%.*]] = load ptr, ptr %Ptr.addr, align 4 // CHECK32-NEXT: call <25 x double> @llvm.matrix.column.major.load.v25f64.i32(ptr align 4 [[PTR]], i32 15, i1 false, i32 5, i32 5) // CHECK64: [[PTR:%.*]] = load ptr, ptr %Ptr.addr, align 8 // CHECK64-NEXT: call <25 x double> @llvm.matrix.column.major.load.v25f64.i64(ptr align 8 [[PTR]], i64 15, i1 false, i32 5, i32 5) dx5x5_t m_a2 = __builtin_matrix_column_major_load(Ptr, 5, 5, 2 * 3 + 9); } void column_major_load_with_variable_stride_ull_float(float *Ptr, unsigned long long S) { // COMMON-LABEL: define{{.*}} void @column_major_load_with_variable_stride_ull_float(ptr %Ptr, i64 %S) // CHECK32: [[S:%.*]] = load i64, ptr %S.addr, align 8 // CHECK32-NEXT: [[STRIDE_TRUNC:%.*]] = trunc i64 [[S]] to i32 // CHECK32-NEXT: [[PTR:%.*]] = load ptr, ptr %Ptr.addr, align 4 // CHECK32-NEXT: call <6 x float> @llvm.matrix.column.major.load.v6f32.i32(ptr align 4 [[PTR]], i32 [[STRIDE_TRUNC]], i1 false, i32 2, i32 3) // CHECK64: [[S:%.*]] = load i64, ptr %S.addr, align 8 // CHECK64-NEXT: [[PTR:%.*]] = load ptr, ptr %Ptr.addr, align 8 // CHECK64-NEXT: call <6 x float> @llvm.matrix.column.major.load.v6f32.i64(ptr align 4 [[PTR]], i64 [[S]], i1 false, i32 2, i32 3) fx2x3_t m_b = __builtin_matrix_column_major_load(Ptr, 2, 3, S); } void column_major_load_with_stride_math_int(int *Ptr, int S) { // COMMON-LABEL: define{{.*}} void @column_major_load_with_stride_math_int(ptr %Ptr, i32 %S) // COMMON: [[S:%.*]] = load i32, ptr %S.addr, align 4 // COMMON-NEXT: [[STRIDE:%.*]] = add nsw i32 [[S]], 32 // CHECK32-NEXT: [[PTR:%.*]] = load ptr, ptr %Ptr.addr, align 4 // CHECK32-NEXT: call <80 x i32> @llvm.matrix.column.major.load.v80i32.i32(ptr align 4 [[PTR]], i32 [[STRIDE]], i1 false, i32 4, i32 20) // // CHECK64-NEXT: [[STRIDE_EXT:%.*]] = sext i32 [[STRIDE]] to i64 // CHECK64-NEXT: [[PTR:%.*]] = load ptr, ptr %Ptr.addr, align 8 // CHECK64-NEXT: call <80 x i32> @llvm.matrix.column.major.load.v80i32.i64(ptr align 4 [[PTR]], i64 [[STRIDE_EXT]], i1 false, i32 4, i32 20) ix4x20_t m_c = __builtin_matrix_column_major_load(Ptr, 4, 20, S + 32); } void column_major_load_with_stride_math_s_int(int *Ptr, short S) { // COMMON-LABEL: define{{.*}} void @column_major_load_with_stride_math_s_int(ptr %Ptr, i16 signext %S) // COMMON: [[S:%.*]] = load i16, ptr %S.addr, align 2 // COMMON-NEXT: [[S_EXT:%.*]] = sext i16 [[S]] to i32 // COMMON-NEXT: [[STRIDE:%.*]] = add nsw i32 [[S_EXT]], 32 // CHECK32-NEXT: [[PTR:%.*]] = load ptr, ptr %Ptr.addr, align 4 // CHECK32-NEXT: %matrix = call <80 x i32> @llvm.matrix.column.major.load.v80i32.i32(ptr align 4 [[PTR]], i32 [[STRIDE]], i1 false, i32 4, i32 20) // // CHECK64-NEXT: [[STRIDE_EXT:%.*]] = sext i32 [[STRIDE]] to i64 // CHECK64-NEXT: [[PTR:%.*]] = load ptr, ptr %Ptr.addr, align 8 // CHECK64-NEXT: %matrix = call <80 x i32> @llvm.matrix.column.major.load.v80i32.i64(ptr align 4 [[PTR]], i64 [[STRIDE_EXT]], i1 false, i32 4, i32 20) ix4x20_t m_c = __builtin_matrix_column_major_load(Ptr, 4, 20, S + 32); } void column_major_load_array1(double Ptr[25]) { // COMMON-LABEL: define{{.*}} void @column_major_load_array1(ptr %Ptr) // CHECK32: [[ADDR:%.*]] = load ptr, ptr %Ptr.addr, align 4 // CHECK32-NEXT: call <25 x double> @llvm.matrix.column.major.load.v25f64.i32(ptr align 4 [[ADDR]], i32 5, i1 false, i32 5, i32 5) // CHECK64: [[ADDR:%.*]] = load ptr, ptr %Ptr.addr, align 8 // CHECK64-NEXT: call <25 x double> @llvm.matrix.column.major.load.v25f64.i64(ptr align 8 [[ADDR]], i64 5, i1 false, i32 5, i32 5) dx5x5_t m = __builtin_matrix_column_major_load(Ptr, 5, 5, 5); } void column_major_load_array2(void) { // COMMON-LABEL: define{{.*}} void @column_major_load_array2() #0 { // COMMON-NEXT: entry: // CHECK32-NEXT: [[PTR:%.*]] = alloca [25 x double], align 8 // CHECK32: [[ARRAY_DEC:%.*]] = getelementptr inbounds [25 x double], ptr [[PTR]], i32 0, i32 0 // CHECK32-NEXT: call <25 x double> @llvm.matrix.column.major.load.v25f64.i32(ptr align 8 [[ARRAY_DEC]], i32 5, i1 false, i32 5, i32 5) // CHECK64-NEXT: [[PTR:%.*]] = alloca [25 x double], align 16 // CHECK64: [[ARRAY_DEC:%.*]] = getelementptr inbounds [25 x double], ptr [[PTR]], i64 0, i64 0 // CHECK64-NEXT: call <25 x double> @llvm.matrix.column.major.load.v25f64.i64(ptr align 16 [[ARRAY_DEC]], i64 5, i1 false, i32 5, i32 5) double Ptr[25]; dx5x5_t m = __builtin_matrix_column_major_load(Ptr, 5, 5, 5); } void column_major_load_const(const double *Ptr) { // COMMON-LABEL: define{{.*}} void @column_major_load_const(ptr %Ptr) // CHECK32: [[PTR:%.*]] = load ptr, ptr %Ptr.addr, align 4 // CHECK32-NEXT: call <25 x double> @llvm.matrix.column.major.load.v25f64.i32(ptr align 4 [[PTR]], i32 5, i1 false, i32 5, i32 5) // // CHECK64: [[PTR:%.*]] = load ptr, ptr %Ptr.addr, align 8 // CHECK64-NEXT: call <25 x double> @llvm.matrix.column.major.load.v25f64.i64(ptr align 8 [[PTR]], i64 5, i1 false, i32 5, i32 5) dx5x5_t m_a1 = __builtin_matrix_column_major_load(Ptr, 5, 5, 5); } void column_major_load_volatile(volatile double *Ptr) { // COMMON-LABEL: define{{.*}} void @column_major_load_volatile(ptr %Ptr) // CHECK32: [[PTR:%.*]] = load ptr, ptr %Ptr.addr, align 4 // CHECK32-NEXT: call <25 x double> @llvm.matrix.column.major.load.v25f64.i32(ptr align 4 [[PTR]], i32 5, i1 true, i32 5, i32 5) // // CHECK64: [[PTR:%.*]] = load ptr, ptr %Ptr.addr, align 8 // CHECK64-NEXT: call <25 x double> @llvm.matrix.column.major.load.v25f64.i64(ptr align 8 [[PTR]], i64 5, i1 true, i32 5, i32 5) dx5x5_t m_a1 = __builtin_matrix_column_major_load(Ptr, 5, 5, 5); } void column_major_store_with_const_stride_double(double *Ptr) { // COMMON-LABEL: define{{.*}} void @column_major_store_with_const_stride_double(ptr %Ptr) // CHECK32: [[M:%.*]] = load <25 x double>, ptr {{.*}}, align 4 // CHECK32-NEXT: [[PTR:%.*]] = load ptr, ptr %Ptr.addr, align 4 // CHECK32-NEXT: call void @llvm.matrix.column.major.store.v25f64.i32(<25 x double> [[M]], ptr align 4 [[PTR]], i32 5, i1 false, i32 5, i32 5) // // CHECK64: [[M:%.*]] = load <25 x double>, ptr {{.*}}, align 8 // CHECK64-NEXT: [[PTR:%.*]] = load ptr, ptr %Ptr.addr, align 8 // CHECK64-NEXT: call void @llvm.matrix.column.major.store.v25f64.i64(<25 x double> [[M]], ptr align 8 [[PTR]], i64 5, i1 false, i32 5, i32 5) dx5x5_t m; __builtin_matrix_column_major_store(m, Ptr, 5); } void column_major_store_with_const_stride2_double(double *Ptr) { // COMMON-LABEL: define{{.*}} void @column_major_store_with_const_stride2_double(ptr %Ptr) // CHECK32: [[M:%.*]] = load <25 x double>, ptr {{.*}}, align 4 // CHECK32-NEXT: [[PTR:%.*]] = load ptr, ptr %Ptr.addr, align 4 // CHECK32-NEXT: call void @llvm.matrix.column.major.store.v25f64.i32(<25 x double> [[M]], ptr align 4 [[PTR]], i32 15, i1 false, i32 5, i32 5) // // CHECK64: [[M:%.*]] = load <25 x double>, ptr {{.*}}, align 8 // CHECK64-NEXT: [[PTR:%.*]] = load ptr, ptr %Ptr.addr, align 8 // CHECK64-NEXT: call void @llvm.matrix.column.major.store.v25f64.i64(<25 x double> [[M]], ptr align 8 [[PTR]], i64 15, i1 false, i32 5, i32 5) // dx5x5_t m; __builtin_matrix_column_major_store(m, Ptr, 2 * 3 + 9); } void column_major_store_with_stride_math_int(int *Ptr, int S) { // COMMON-LABEL: define{{.*}} void @column_major_store_with_stride_math_int(ptr %Ptr, i32 %S) // COMMON: [[M:%.*]] = load <80 x i32>, ptr {{.*}}, align 4 // CHECK32-NEXT: [[PTR:%.*]] = load ptr, ptr %Ptr.addr, align 4 // CHECK64-NEXT: [[PTR:%.*]] = load ptr, ptr %Ptr.addr, align 8 // COMMON-NEXT: [[S:%.*]] = load i32, ptr %S.addr, align 4 // COMMON-NEXT: [[ADD:%.*]] = add nsw i32 [[S]], 32 // CHECK32-NEXT: call void @llvm.matrix.column.major.store.v80i32.i32(<80 x i32> [[M]], ptr align 4 [[PTR]], i32 [[ADD]], i1 false, i32 4, i32 20) // // CHECK64-NEXT: [[IDX:%.*]] = sext i32 [[ADD]] to i64 // CHECK64-NEXT: call void @llvm.matrix.column.major.store.v80i32.i64(<80 x i32> [[M]], ptr align 4 [[PTR]], i64 [[IDX]], i1 false, i32 4, i32 20) ix4x20_t m; __builtin_matrix_column_major_store(m, Ptr, S + 32); } void column_major_store_with_stride_math_s_int(int *Ptr, short S) { // COMMON-LABEL: define{{.*}} void @column_major_store_with_stride_math_s_int(ptr %Ptr, i16 signext %S) // COMMON: [[M:%.*]] = load <80 x i32>, ptr {{.*}}, align 4 // CHECK32-NEXT: [[PTR:%.*]] = load ptr, ptr %Ptr.addr, align 4 // CHECK64-NEXT: [[PTR:%.*]] = load ptr, ptr %Ptr.addr, align 8 // COMMON-NEXT: [[S:%.*]] = load i16, ptr %S.addr, align 2 // COMMON-NEXT: [[EXT:%.*]] = sext i16 [[S]] to i32 // COMMON-NEXT: [[ADD:%.*]] = add nsw i32 [[EXT]], 2 // CHECK32-NEXT: call void @llvm.matrix.column.major.store.v80i32.i32(<80 x i32> [[M]], ptr align 4 [[PTR]], i32 [[ADD]], i1 false, i32 4, i32 20) // // CHECK64-NEXT: [[IDX:%.*]] = sext i32 [[ADD]] to i64 // CHECK64-NEXT: call void @llvm.matrix.column.major.store.v80i32.i64(<80 x i32> [[M]], ptr align 4 [[PTR]], i64 [[IDX]], i1 false, i32 4, i32 20) ix4x20_t m; __builtin_matrix_column_major_store(m, Ptr, S + 2); } void column_major_store_array1(double Ptr[25]) { // COMMON-LABEL: define{{.*}} void @column_major_store_array1(ptr %Ptr) // CHECK32: [[M:%.*]] = load <25 x double>, ptr {{.*}}, align 4 // CHECK32-NEXT: [[PTR:%.*]] = load ptr, ptr %Ptr.addr, align 4 // CHECK32-NEXT: call void @llvm.matrix.column.major.store.v25f64.i32(<25 x double> [[M]], ptr align 4 [[PTR]], i32 5, i1 false, i32 5, i32 5) // // CHECK64: [[M:%.*]] = load <25 x double>, ptr {{.*}}, align 8 // CHECK64-NEXT: [[PTR:%.*]] = load ptr, ptr %Ptr.addr, align 8 // CHECK64-NEXT: call void @llvm.matrix.column.major.store.v25f64.i64(<25 x double> [[M]], ptr align 8 [[PTR]], i64 5, i1 false, i32 5, i32 5) dx5x5_t m; __builtin_matrix_column_major_store(m, Ptr, 5); } void column_major_store_array2(void) { // COMMON-LABEL: define{{.*}} void @column_major_store_array2() // CHECK32: [[M:%.*]] = load <25 x double>, ptr {{.*}}, align 4 // CHECK32-NEXT: [[PTR:%.*]] = getelementptr inbounds [25 x double], ptr %Ptr, i32 0, i32 0 // CHECK32-NEXT: call void @llvm.matrix.column.major.store.v25f64.i32(<25 x double> [[M]], ptr align 8 [[PTR]], i32 5, i1 false, i32 5, i32 5) // // CHECK64: [[M:%.*]] = load <25 x double>, ptr {{.*}}, align 8 // CHECK64-NEXT: [[PTR:%.*]] = getelementptr inbounds [25 x double], ptr %Ptr, i64 0, i64 0 // CHECK64-NEXT: call void @llvm.matrix.column.major.store.v25f64.i64(<25 x double> [[M]], ptr align 16 [[PTR]], i64 5, i1 false, i32 5, i32 5) double Ptr[25]; dx5x5_t m; __builtin_matrix_column_major_store(m, Ptr, 5); } void column_major_store_volatile(volatile double *Ptr) { // COMMON-LABEL: define{{.*}} void @column_major_store_volatile(ptr %Ptr) #0 { // CHECK32: [[M:%.*]] = load <25 x double>, ptr {{.*}}, align 4 // CHECK32-NEXT: [[PTR:%.*]] = load ptr, ptr %Ptr.addr, align 4 // CHECK32-NEXT: call void @llvm.matrix.column.major.store.v25f64.i32(<25 x double> [[M]], ptr align 4 [[PTR]], i32 5, i1 true, i32 5, i32 5) // // CHECK64: [[M:%.*]] = load <25 x double>, ptr {{.*}}, align 8 // CHECK64-NEXT: [[PTR:%.*]] = load ptr, ptr %Ptr.addr, align 8 // CHECK64-NEXT: call void @llvm.matrix.column.major.store.v25f64.i64(<25 x double> [[M]], ptr align 8 [[PTR]], i64 5, i1 true, i32 5, i32 5) dx5x5_t m; __builtin_matrix_column_major_store(m, Ptr, 5); }