minor changes

HanGuo97 · Dec 11, 2024 · 6a80d6f · 6a80d6f
1 parent f0cdec6
commit 6a80d6f
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 6 deletions.
diff --git a/.github/workflows/wheels.yaml b/.github/workflows/wheels.yaml
@@ -23,6 +23,8 @@ jobs:
     steps:
       - name: Checkout
         uses: actions/checkout@v4
+        with:
+          submodules: recursive
 
       - name: Set up Linux Env
         if: ${{ runner.os == 'Linux' }}

diff --git a/flute/tune.py b/flute/tune.py
@@ -377,6 +377,7 @@ def tune_and_pack(
     group_size: int,
     num_seeds: int = 3,
     check_correctness: bool = True,
+    check_num_seeds: int = 3,
 ) -> Tuple[torch.Tensor, TuneMetaData]:
     if inputs.ndim != 2:
         raise ValueError
@@ -422,12 +423,14 @@ def tune_and_pack(
     if check_correctness is True:
         for uniform in [True, False]:
             for identity in [True, False]:
-                check(
-                    weight=weight,
-                    weight_packed=weight_packed,
-                    metadata=metadata,
-                    uniform=uniform,
-                    identity=identity)
+                for seed in range(check_num_seeds):
+                    torch.manual_seed(seed)
+                    check(
+                        weight=weight,
+                        weight_packed=weight_packed,
+                        metadata=metadata,
+                        uniform=uniform,
+                        identity=identity)
 
     return weight_packed, metadata