udpate tests

mila-iqia · Aug 1, 2024 · dc915c7 · dc915c7
1 parent 7bd5f0c
commit dc915c7
Show file tree

Hide file tree

Showing 11 changed files with 549 additions and 385 deletions.
diff --git a/.pin/constraints-cuda-torch.txt b/.pin/constraints-cuda-torch.txt
diff --git a/benchmarks/llm/requirements.cuda.txt b/benchmarks/llm/requirements.cuda.txt
diff --git a/benchmate/benchmate/benchrun.py b/benchmate/benchmate/benchrun.py
@@ -64,9 +64,8 @@ def run(args):
 
 
 def main(args=None):
-    with forward_voir_file():
-        distrun.torchrun(args)
-
+    args = distrun.parse_args(args)
+    run(args)
 
 
 if __name__ == "__main__":

diff --git a/config/standard.yaml b/config/standard.yaml
@@ -73,10 +73,6 @@ super-slomo:
   enabled: true
   weight: 1.0
 
-dlrm:
-  enabled: true
-  weight: 1.0
-
 fp16:
   enabled: true
   weight: 0.0

diff --git a/milabench/_version.py b/milabench/_version.py
@@ -1,5 +1,5 @@
 """This file is generated, do not modify"""
 
-__tag__ = "v0.1.0-24-g2586e650"
-__commit__ = "2586e65040275fc31aabd2f5de100154ea700e92"
-__date__ = "2024-07-26 15:26:09 -0400"
+__tag__ = "v0.1.0-34-g7bd5f0c7"
+__commit__ = "7bd5f0c7e7ff78e6c44d54580ef376473f583396"
+__date__ = "2024-08-01 13:13:19 -0400"