Skip to content

[regression test] Update run_mantaray_jobs.py for splited test order for PyTorch regression test on TPU #572

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
ManfeiBai wants to merge 13 commits into
base: master
Choose a base branch
from
Open

[regression test] Update run_mantaray_jobs.py for splited test order for PyTorch regression test on TPU #572

Changes from all commits
Commits
Show all changes
13 commits
Select commit Hold shift + click to select a range
6d03959
Update run_mantaray_jobs.py for splited dag files
ManfeiBai Jan 21, 2025
d52b1a7
Update run_mantaray_jobs.py
ManfeiBai Jan 22, 2025
70bb7f2
Update run_mantaray_jobs.py
ManfeiBai Jan 22, 2025
80b09ec
Update run_mantaray_jobs.py
ManfeiBai Jan 22, 2025
9dd0cc4
Update run_mantaray_jobs.py
ManfeiBai Jan 22, 2025
355c184
Update run_mantaray_jobs.py
ManfeiBai Jan 22, 2025
2043f2f
Update run_mantaray_jobs.py
ManfeiBai Jan 22, 2025
029cd48
Update run_mantaray_jobs.py
ManfeiBai Jan 22, 2025
d054742
Update run_mantaray_jobs.py
ManfeiBai Jan 22, 2025
a136449
Update run_mantaray_jobs.py
ManfeiBai Jan 22, 2025
9f60fbf
Merge branch 'master' into ManfeiBai-patch-8
ManfeiBai Jan 22, 2025
06f2227
Update run_mantaray_jobs.py
ManfeiBai Jan 24, 2025
2e0d4df
Update run_mantaray_jobs.py
ManfeiBai Jan 24, 2025
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 2 additions & 1 deletion dags/mantaray/run_mantaray_jobs.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -41,10 +41,11 @@
# merge all PyTorch/XLA tests ino one Dag
with models.DAG(
dag_id="pytorch_xla_model_regression_test_on_trillium",
schedule="0 0 * * *", # everyday at midnight # job["schedule"],
schedule="0 0 * * *", # everyday at midnight
tags=["mantaray", "pytorchxla", "xlml"],
start_date=datetime.datetime(2024, 4, 22),
catchup=False,
concurrency=1,
) as dag:
for workload_file_name in workload_file_name_list:
run_workload = mantaray.run_workload.override(
Expand Down
Loading