Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/c10/cuda/
H A DCUDAAllocatorConfig.cpp199 bool& used_cudaMallocAsync) { in parseAllocatorConfig() argument
206 used_cudaMallocAsync = (config[i] == "cudaMallocAsync"); in parseAllocatorConfig()
209 if (used_cudaMallocAsync) { in parseAllocatorConfig()
242 bool used_cudaMallocAsync = false; in parseArgs() local
268 i = parseAllocatorConfig(config, i, used_cudaMallocAsync); in parseArgs()
319 if (used_cudaMallocAsync && used_native_specific_option) { in parseArgs()
H A DCUDAAllocatorConfig.h102 bool& used_cudaMallocAsync);