Save other CI jobs' result (torch/tf pipeline, example, deepspeed etc) (#30699)

* update * update * update * update * update * update * update * update * Update utils/notification_service.py Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com> --------- Co-authored-by: ydshieh <ydshieh@users.noreply.github.com> Co-authored-by: amyeroberts <22614925+amyeroberts@users.noreply.github.com>
2024-05-13 17:27:44 +02:00
parent 2e27291ce4
commit 82c1625ec3
3 changed files with 46 additions and 17 deletions
--- a/utils/notification_service_quantization.py
+++ b/utils/notification_service_quantization.py
@@ -242,6 +242,13 @@ if __name__ == "__main__":
                            {"line": line, "trace": stacktraces.pop(0)}
                        )

+    job_name = os.getenv("CI_TEST_JOB")
+    if not os.path.isdir(os.path.join(os.getcwd(), f"ci_results_{job_name}")):
+        os.makedirs(os.path.join(os.getcwd(), f"ci_results_{job_name}"))
+
+    with open(f"ci_results_{job_name}/quantization_results.json", "w", encoding="UTF-8") as fp:
+        json.dump(quantization_results, fp, indent=4, ensure_ascii=False)
+
    message = QuantizationMessage(
        title,
        results=quantization_results,