| 19 | namespace gpu { |
| 20 | |
| 21 | CUfunc_cache GpuKernel::GetGpuCacheConfig() const { |
| 22 | switch (preferred_cache_config_) { |
| 23 | case KernelCacheConfig::kNoPreference: |
| 24 | return CU_FUNC_CACHE_PREFER_NONE; |
| 25 | case KernelCacheConfig::kPreferShared: |
| 26 | return CU_FUNC_CACHE_PREFER_SHARED; |
| 27 | case KernelCacheConfig::kPreferL1: |
| 28 | return CU_FUNC_CACHE_PREFER_L1; |
| 29 | case KernelCacheConfig::kPreferEqual: |
| 30 | return CU_FUNC_CACHE_PREFER_EQUAL; |
| 31 | default: |
| 32 | LOG(FATAL) << "Unknown KernelCacheConfig" |
| 33 | << static_cast<int32>(preferred_cache_config_); |
| 34 | } |
| 35 | } |
| 36 | |
| 37 | } // namespace gpu |
| 38 | } // namespace stream_executor |