diff --git a/aten/src/ATen/cuda/tunable/Tunable.cpp b/aten/src/ATen/cuda/tunable/Tunable.cpp index 1ef425b617c3a..ff45d43b3d8b3 100644 --- a/aten/src/ATen/cuda/tunable/Tunable.cpp +++ b/aten/src/ATen/cuda/tunable/Tunable.cpp @@ -458,6 +458,8 @@ void TuningContext::EnableRecordUntuned(bool value) { TUNABLE_LOG1("Enable Record Untuned for TunableOp"); } else { TUNABLE_LOG1("Disable Record Untuned for TunableOp"); + TUNABLE_LOG1("Closing Untuned GEMM Results File"); + untuned_file_.close(); } }