Skip to content

Commit 6771315

Browse files
authored
Update SYCL_INTEL_bf16_conversion.asciidoc
1 parent 64c375f commit 6771315

File tree

1 file changed

+15
-15
lines changed

1 file changed

+15
-15
lines changed

sycl/doc/extensions/Bf16Conversion/SYCL_INTEL_bf16_conversion.asciidoc

Lines changed: 15 additions & 15 deletions
Original file line numberDiff line numberDiff line change
@@ -111,7 +111,8 @@ namespace ext {
111111
namespace intel {
112112
namespace experimental {
113113
114-
class bfloat16 {
114+
class [[sycl_detail::uses_aspects(ext_intel_bf16_conversion)]]
115+
bfloat16 {
115116
using storage_t = uint16_t;
116117
storage_t value;
117118
@@ -156,21 +157,20 @@ float foo(float a, float b) {
156157
return C;
157158
}
158159
159-
int main (int argc, char *argv[])
160-
{
161-
float data[3] = {7.0, 8.1, 0.0};
162-
cl::sycl::queue deviceQueue;
163-
cl::sycl::buffer<float, 1> buf {data, cl::sycl::range<1> {3}};
164-
165-
deviceQueue.submit ([&] (cl::sycl::handler& cgh)
166-
{
167-
auto numbers = buf.get_access<cl::sycl::access::mode::read_write> (cgh);
168-
cgh.single_task<class simple_kernel> ([=] ()
169-
{
170-
numbers[2] = foo(numbers[0], numbers[1]);
171-
172-
});
160+
int main (int argc, char *argv[]) {
161+
float data[3] = {7.0, 8.1, 0.0};
162+
sycl::device dev{sycl::default_selector{}};
163+
sycl::queue deviceQueue{dev};
164+
sycl::buffer<float, 1> buf {data, cl::sycl::range<1> {3}};
165+
166+
if (dev.has(aspect::ext_intel_bf16_conversion)) {
167+
deviceQueue.submit ([&] (cl::sycl::handler& cgh) {
168+
auto numbers = buf.get_access<cl::sycl::access::mode::read_write> (cgh);
169+
cgh.single_task<class simple_kernel> ([=] () {
170+
numbers[2] = foo(numbers[0], numbers[1]);
171+
});
173172
});
173+
}
174174
return 0;
175175
}
176176
----

0 commit comments

Comments
 (0)