Home
last modified time | relevance | path

Searched refs:cache_value (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dnvptx_compiler.cc352 CompilationCacheValue* cache_value = nullptr; in CompileGpuAsmOrGetCachedResult() local
361 cache_value = &iter->second; in CompileGpuAsmOrGetCachedResult()
368 tensorflow::mutex_lock lock(cache_value->mutex_); in CompileGpuAsmOrGetCachedResult()
370 CHECK(!cache_value->compilation_done); in CompileGpuAsmOrGetCachedResult()
380 cache_value->cubin_data = std::move(maybe_cubin).ValueOrDie(); in CompileGpuAsmOrGetCachedResult()
382 << " CUBIN size: " << cache_value->cubin_data.size(); in CompileGpuAsmOrGetCachedResult()
432 cache_value->compilation_done = true; in CompileGpuAsmOrGetCachedResult()
433 cache_value->compilation_done_cv_.notify_all(); in CompileGpuAsmOrGetCachedResult()
435 while (!cache_value->compilation_done) { in CompileGpuAsmOrGetCachedResult()
436 cache_value->compilation_done_cv_.wait(lock); in CompileGpuAsmOrGetCachedResult()
[all …]