torch.compiler.save_cache_artifacts

torch.compiler.save_cache_artifacts()

Serializes all the cache artifacts that were created during the compilation

Example:

• Execute torch.compile

• Call torch.compiler.save_cache_artifacts()

Return type:

tuple[bytes, CacheInfo] | None