Skip to content

Commit cb42f35

Browse files
More robust checking for fence variable signaling extension is present
1 parent cf8d89a commit cb42f35

File tree

1 file changed

+14
-5
lines changed

1 file changed

+14
-5
lines changed

dpctl/tensor/libtensor/include/utils/sycl_utils.hpp

Lines changed: 14 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -423,7 +423,8 @@ struct Identity<Op, T, std::enable_if_t<UseBuiltInIdentity<Op, T>::value>>
423423

424424
// Sub-group load/store
425425

426-
#if defined(SYCL_EXT_ONEAPI_GROUP_LOAD_STORE)
426+
#if defined(SYCL_EXT_ONEAPI_GROUP_LOAD_STORE) && \
427+
(SYCL_EXT_ONEAPI_GROUP_LOAD_STORE)
427428
namespace ls_ns = sycl::ext::oneapi::experimental;
428429
#endif
429430

@@ -434,7 +435,8 @@ template <std::uint8_t vec_sz,
434435
auto sub_group_load(const sycl::sub_group &sg,
435436
sycl::multi_ptr<ElementType, Space, DecorateAddress> m_ptr)
436437
{
437-
#if defined(SYCL_EXT_ONEAPI_GROUP_LOAD_STORE)
438+
#if defined(SYCL_EXT_ONEAPI_GROUP_LOAD_STORE) && \
439+
(SYCL_EXT_ONEAPI_GROUP_LOAD_STORE)
438440
sycl::vec<ElementType, vec_sz> x;
439441
ls_ns::group_load(sg, m_ptr, x);
440442
return x;
@@ -449,7 +451,8 @@ template <sycl::access::address_space Space,
449451
auto sub_group_load(const sycl::sub_group &sg,
450452
sycl::multi_ptr<ElementType, Space, DecorateAddress> m_ptr)
451453
{
452-
#if defined(SYCL_EXT_ONEAPI_GROUP_LOAD_STORE)
454+
#if defined(SYCL_EXT_ONEAPI_GROUP_LOAD_STORE) && \
455+
(SYCL_EXT_ONEAPI_GROUP_LOAD_STORE)
453456
ElementType x;
454457
ls_ns::group_load(sg, m_ptr, x);
455458
return x;
@@ -466,10 +469,13 @@ void sub_group_store(const sycl::sub_group &sg,
466469
const sycl::vec<ElementType, vec_sz> &val,
467470
sycl::multi_ptr<ElementType, Space, DecorateAddress> m_ptr)
468471
{
469-
#if defined(SYCL_EXT_ONEAPI_GROUP_LOAD_STORE)
472+
#if defined(SYCL_EXT_ONEAPI_GROUP_LOAD_STORE) && \
473+
(SYCL_EXT_ONEAPI_GROUP_LOAD_STORE)
470474
ls_ns::group_store(sg, val, m_ptr);
475+
return;
471476
#else
472477
sg.store<vec_sz>(m_ptr, val);
478+
return;
473479
#endif
474480
}
475481

@@ -480,10 +486,13 @@ void sub_group_store(const sycl::sub_group &sg,
480486
const ElementType &val,
481487
sycl::multi_ptr<ElementType, Space, DecorateAddress> m_ptr)
482488
{
483-
#if defined(SYCL_EXT_ONEAPI_GROUP_LOAD_STORE)
489+
#if defined(SYCL_EXT_ONEAPI_GROUP_LOAD_STORE) && \
490+
(SYCL_EXT_ONEAPI_GROUP_LOAD_STORE)
484491
ls_ns::group_store(sg, val, m_ptr);
492+
return;
485493
#else
486494
sg.store(m_ptr, val);
495+
return;
487496
#endif
488497
}
489498

0 commit comments

Comments
 (0)