MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / FuncSetCacheConfig

Method FuncSetCacheConfig

tensorflow/stream_executor/cuda/cuda_driver.cc:493–503  ·  view source on GitHub ↗

static */

Source from the content-addressed store, hash-verified

491}
492
493/* static */ bool GpuDriver::FuncSetCacheConfig(CUfunction function,
494 CUfunc_cache cache_config) {
495 CUresult res = cuFuncSetCacheConfig(function, cache_config);
496 if (res != CUDA_SUCCESS) {
497 LOG(ERROR) << "failed to set CUDA kernel cache config. kernel: " << function
498 << ", config: " << cache_config << ", result: " << ToString(res);
499 return false;
500 }
501
502 return true;
503}
504
505/* static */ port::StatusOr<CUsharedconfig>
506GpuDriver::ContextGetSharedMemConfig(GpuContext* context) {

Callers

nothing calls this directly

Calls 1

ToStringFunction · 0.70

Tested by

no test coverage detected