Searched refs:driver_allocated_memory (Results 1 – 3 of 3) sorted by relevance
19 driver_allocated_memory
123 def driver_allocated_memory() -> int: function
1148 self.driver_before = torch.mps.driver_allocated_memory()1173 driver_mem_allocated = torch.mps.driver_allocated_memory()1285 driver_before = torch.mps.driver_allocated_memory()1288 driver_after = torch.mps.driver_allocated_memory()8013 driver_alloc_before = torch.mps.driver_allocated_memory()8018 driver_alloc_after = torch.mps.driver_allocated_memory()