shylie/llvm-project
1
0
Fork 0
Code Issues Pull Requests Actions 6 Packages Projects Releases Wiki Activity

[Clang] Support __bf16 type for SPIR/SPIR-V (#169012)

Browse Source 
SPIR/SPIR-V are generic targets. Assume they support __bf16.
This commit is contained in:
Wenju He 2025-11-24 10:10:11 +08:00 committed by GitHub
parent fe56f5c3d3
commit c4254cd9bb
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
3 changed files with 39 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
clang/lib/Basic/Targets/SPIR.h
View File

@ -108,6 +108,10 @@ protected:
UseAddrSpaceMapMangling = true;
HasFastHalfType = true;
HasFloat16 = true;
HasBFloat16 = true;
HasFullBFloat16 = true;
BFloat16Width = BFloat16Align = 16;
BFloat16Format = &llvm::APFloat::BFloat();
// Define available target features
// These must be defined in sorted order!
NoAsmVariants = true;
@ -427,9 +431,6 @@ public:
resetDataLayout("e-i64:64-v16:16-v24:32-v32:32-v48:64-v96:128-v192:256-"
"v256:256-v512:512-v1024:1024-n32:64-S32-G1-P4-A0");
BFloat16Width = BFloat16Align = 16;
BFloat16Format = &llvm::APFloat::BFloat();
HasFastHalfType = true;
HasFloat16 = true;
HalfArgsAndReturns = true;
@ -437,8 +438,6 @@ public:
MaxAtomicPromoteWidth = MaxAtomicInlineWidth = 64;
}
bool hasBFloat16Type() const override { return true; }
ArrayRef<const char *> getGCCRegNames() const override;
BuiltinVaListKind getBuiltinVaListKind() const override {

31
clang/test/CodeGenOpenCL/__bf16.cl Normal file
View File

@ -0,0 +1,31 @@
// RUN: %clang_cc1 %s -cl-std=cl3.0 -emit-llvm -o - -triple spir-unknown-unknown | FileCheck %s
// RUN: %clang_cc1 %s -cl-std=cl3.0 -emit-llvm -o - -triple spirv64-unknown-unknown | FileCheck %s
kernel void test(global __bf16 *a, global __bf16 *b){
// CHECK-LABEL: spir_kernel void @test(
// CHECK: fadd bfloat
// CHECK: fsub bfloat
// CHECK: fmul bfloat
// CHECK: fdiv bfloat
*b += *a;
*b -= *a;
*b *= *a;
*b /= *a;
}
typedef __bf16 __bf16v4 __attribute((ext_vector_type(4)));
kernel void test_v4(global __bf16v4 *a, global __bf16v4 *b){
// CHECK-LABEL: spir_kernel void @test_v4(
// CHECK: fadd <4 x bfloat>
// CHECK: fsub <4 x bfloat>
// CHECK: fmul <4 x bfloat>
// CHECK: fdiv <4 x bfloat>
*b += *a;
*b -= *a;
*b *= *a;
*b /= *a;
}

7
clang/test/SemaSYCL/bf16.cpp
View File

@ -1,8 +1,9 @@
// RUN: %clang_cc1 -triple spir64 -aux-triple x86_64-unknown-linux-gnu -fsycl-is-device -verify -fsyntax-only %s
// expected-no-diagnostics
template <typename Name, typename Func>
__attribute__((sycl_kernel)) void kernel(const Func &kernelFunc) {
kernelFunc(); // expected-note {{called by 'kernel}}
kernelFunc();
}
void host_ok(void) {
@ -11,9 +12,9 @@ void host_ok(void) {
int main()
{ host_ok();
__bf16 var; // expected-note {{'var' defined here}}
__bf16 var;
kernel<class variables>([=]() {
(void)var; // expected-error {{'var' requires 16 bit size '__bf16' type support, but target 'spir64' does not support it}}
(void)var;
int B = sizeof(__bf16);
});