diff --git a/clang/lib/Headers/amxintrin.h b/clang/lib/Headers/amxintrin.h --- a/clang/lib/Headers/amxintrin.h +++ b/clang/lib/Headers/amxintrin.h @@ -267,8 +267,8 @@ } static __inline__ _tile1024i __DEFAULT_FN_ATTRS_BF16 -_tile_tdpbf16ps_internal(unsigned short m, unsigned short n, unsigned short k, - _tile1024i dst, _tile1024i src1, _tile1024i src2) { +_tile_dpbf16ps_internal(unsigned short m, unsigned short n, unsigned short k, + _tile1024i dst, _tile1024i src1, _tile1024i src2) { return __builtin_ia32_tdpbf16ps_internal(m, n, k, dst, src1, src2); } @@ -323,10 +323,10 @@ } __DEFAULT_FN_ATTRS_BF16 -static void __tile_tdpbf16ps(__tile1024i *dst, __tile1024i src1, - __tile1024i src2) { - dst->tile = _tile_tdpbf16ps_internal(src1.row, src2.col, src1.col, dst->tile, - src1.tile, src2.tile); +static void __tile_dpbf16ps(__tile1024i *dst, __tile1024i src1, + __tile1024i src2) { + dst->tile = _tile_dpbf16ps_internal(src1.row, src2.col, src1.col, dst->tile, + src1.tile, src2.tile); } #undef __DEFAULT_FN_ATTRS_TILE diff --git a/clang/test/CodeGen/X86/amx_api.c b/clang/test/CodeGen/X86/amx_api.c --- a/clang/test/CodeGen/X86/amx_api.c +++ b/clang/test/CodeGen/X86/amx_api.c @@ -81,9 +81,9 @@ __tile_zero(&c); } -void test_tile_tdpbf16ps(__tile1024i a, __tile1024i b, __tile1024i c) { - //CHECK-LABEL: @test_tile_tdpbf16ps +void test_tile_dpbf16ps(__tile1024i a, __tile1024i b, __tile1024i c) { + //CHECK-LABEL: @test_tile_dpbf16ps //CHECK: call x86_amx @llvm.x86.tdpbf16ps.internal //CHECK-NEXT: {{%.*}} = bitcast x86_amx {{%.*}} to <256 x i32> - __tile_tdpbf16ps(&a, b, c); + __tile_dpbf16ps(&a, b, c); }