Searched refs:get_gpu_shared_memory (Results 1 – 5 of 5) sorted by relevance
13 from torch._inductor.utils import get_gpu_shared_memory, is_big_gpu111 and get_gpu_shared_memory() == 166912116 and get_gpu_shared_memory() == 232448
11 from torch._inductor.utils import get_gpu_shared_memory101 shared_memory = get_gpu_shared_memory()
22 from torch._inductor.utils import get_gpu_shared_memory86 get_gpu_shared_memory(),
33 get_gpu_shared_memory,544 ) or get_gpu_shared_memory() != 166912:
1542 def get_gpu_shared_memory(): function