enable_log_tensor_save_tensors_to_disk

torchao.prototype.quant_logger.enable_log_tensor_save_tensors_to_disk(save_dir)

Redefines quant_logger::log_tensor to save full tensors to disk.

Each logged tensor is cloned and saved as {fqn}_{op}_{tag}.pt under save_dir. The directory is created if it does not exist.

Parameters:

save_dir – Path to the directory where tensor files will be saved.