fix(tracker): reduction pt per layer

Author: OneAndZero24

activationscope/tracker.py

@@ -183,13 +183,15 @@ def register_reduction(
         if dummy_acc is None:
             dummy_acc = torch.randn(64, dtype=torch.float32)
 
-        reduction_path = export_reduction(fn, dummy_acc, dummy_tensor)
-        self._temp_files.append(reduction_path)
-
         if layers is None:
+            reduction_path = export_reduction(fn, dummy_acc, dummy_tensor)
+            self._temp_files.append(reduction_path)
             self._global_reduction_path = reduction_path
         else:
-            self._reductions.append((reduction_path, layers))
+            for layer in layers:
+                reduction_path = export_reduction(fn, dummy_acc, dummy_tensor)
+                self._temp_files.append(reduction_path)
+                self._reductions.append((reduction_path, [layer]))
 
     # ── Built-in reduction factories ──────────────────────────────
 

utils/generate-pyproject.py

@@ -46,7 +46,7 @@ def generate(template: str, combos: List[Dict[str, str]]) -> str:
     text = text.replace("__TORCH_VERSION__", torch_floor)
     text = text.replace(
         "__VERSION__",
-        os.environ.get("ASCOPE_VERSION", "0.1.0"),
+        os.environ.get("ASCOPE_VERSION", "0.1.1"),
     )
     text = text.replace("__PYTHON_CLASSIFIERS__", classifiers + ",\n")