Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/cuda/
H A Dmemory_snapshot.cpp367 IValue release_lock_on_malloc_s = "release_lock_on_cudamalloc"; in _memory_snapshot_pickled() local
385 release_lock_on_malloc_s, in _memory_snapshot_pickled()
H A DModule.cpp818 py::str release_lock_on_malloc_s = "release_lock_on_cudamalloc"; in THCPModule_memorySnapshot() local
832 allocator_settings[release_lock_on_malloc_s] = in THCPModule_memorySnapshot()