torch.mps.profiler.is_metal_capture_enabled

torch.mps.profiler.is_metal_capture_enabled() [source]

Checks if metal_capture context manager is usable To enable metal capture, set MTL_CAPTURE_ENABLED envvar

Return type

bool