Skip to content

Commit efb00dd

Browse files
committed
Move MKernelCacheConfig to KernelData
1 parent 2a5d5eb commit efb00dd

File tree

3 files changed

+17
-7
lines changed

3 files changed

+17
-7
lines changed

sycl/source/detail/handler_impl.hpp

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -104,8 +104,6 @@ class handler_impl {
104104
// If the pipe operation is read or write, 1 for read 0 for write.
105105
bool HostPipeRead = true;
106106

107-
ur_kernel_cache_config_t MKernelCacheConfig = UR_KERNEL_CACHE_CONFIG_DEFAULT;
108-
109107
// Extra information for bindless image copy
110108
ur_image_desc_t MSrcImageDesc = {};
111109
ur_image_desc_t MDstImageDesc = {};

sycl/source/detail/kernel_data.hpp

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -115,6 +115,14 @@ class KernelData {
115115

116116
bool usesAssert() const { return MDeviceKernelInfoPtr->usesAssert(); }
117117

118+
ur_kernel_cache_config_t getKernelCacheConfig() const {
119+
return MKernelCacheConfig;
120+
}
121+
122+
void setKernelCacheConfig(ur_kernel_cache_config_t Config) {
123+
MKernelCacheConfig = Config;
124+
}
125+
118126
bool isCooperative() const { return MKernelIsCooperative; }
119127

120128
void setCooperative(bool IsCooperative) {
@@ -152,6 +160,8 @@ class KernelData {
152160
/// The list of arguments for the kernel.
153161
std::vector<detail::ArgDesc> MArgs;
154162

163+
ur_kernel_cache_config_t MKernelCacheConfig = UR_KERNEL_CACHE_CONFIG_DEFAULT;
164+
155165
bool MKernelIsCooperative = false;
156166
bool MKernelUsesClusterLaunch = false;
157167
uint32_t MKernelWorkGroupMemorySize = 0;

sycl/source/handler.cpp

Lines changed: 7 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -663,7 +663,8 @@ event handler::finalize() {
663663
impl->MKernelData.getArgs(), KernelBundleImpPtr,
664664
MKernel.get(), toKernelNameStrT(MKernelName),
665665
*impl->MKernelData.getDeviceKernelInfoPtr(), RawEvents,
666-
ResultEvent.get(), nullptr, impl->MKernelCacheConfig,
666+
ResultEvent.get(), nullptr,
667+
impl->MKernelData.getKernelCacheConfig(),
667668
impl->MKernelData.isCooperative(),
668669
impl->MKernelData.usesClusterLaunch(),
669670
impl->MKernelData.getKernelWorkGroupMemorySize(),
@@ -731,7 +732,8 @@ event handler::finalize() {
731732
toKernelNameStrT(MKernelName),
732733
*impl->MKernelData.getDeviceKernelInfoPtr(), std::move(MStreamStorage),
733734
std::move(impl->MAuxiliaryResources), getType(),
734-
impl->MKernelCacheConfig, impl->MKernelData.isCooperative(),
735+
impl->MKernelData.getKernelCacheConfig(),
736+
impl->MKernelData.isCooperative(),
735737
impl->MKernelData.usesClusterLaunch(),
736738
impl->MKernelData.getKernelWorkGroupMemorySize(), MCodeLoc));
737739
break;
@@ -2011,13 +2013,13 @@ detail::context_impl &handler::getContextImpl() const {
20112013
void handler::setKernelCacheConfig(handler::StableKernelCacheConfig Config) {
20122014
switch (Config) {
20132015
case handler::StableKernelCacheConfig::Default:
2014-
impl->MKernelCacheConfig = UR_KERNEL_CACHE_CONFIG_DEFAULT;
2016+
impl->MKernelData.setKernelCacheConfig(UR_KERNEL_CACHE_CONFIG_DEFAULT);
20152017
break;
20162018
case handler::StableKernelCacheConfig::LargeSLM:
2017-
impl->MKernelCacheConfig = UR_KERNEL_CACHE_CONFIG_LARGE_SLM;
2019+
impl->MKernelData.setKernelCacheConfig(UR_KERNEL_CACHE_CONFIG_LARGE_SLM);
20182020
break;
20192021
case handler::StableKernelCacheConfig::LargeData:
2020-
impl->MKernelCacheConfig = UR_KERNEL_CACHE_CONFIG_LARGE_DATA;
2022+
impl->MKernelData.setKernelCacheConfig(UR_KERNEL_CACHE_CONFIG_LARGE_DATA);
20212023
break;
20222024
}
20232025
}

0 commit comments

Comments
 (0)