alloc 0.0.11__tar.gz → 0.0.12__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {alloc-0.0.11 → alloc-0.0.12}/PKG-INFO +1 -1
- {alloc-0.0.11 → alloc-0.0.12}/pyproject.toml +1 -1
- {alloc-0.0.11 → alloc-0.0.12}/src/alloc/__init__.py +1 -1
- {alloc-0.0.11 → alloc-0.0.12}/src/alloc/probe.py +10 -0
- {alloc-0.0.11 → alloc-0.0.12}/src/alloc.egg-info/PKG-INFO +1 -1
- {alloc-0.0.11 → alloc-0.0.12}/README.md +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/setup.cfg +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/src/alloc/artifact_loader.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/src/alloc/artifact_writer.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/src/alloc/browser_auth.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/src/alloc/callbacks.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/src/alloc/catalog/__init__.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/src/alloc/catalog/default_rate_card.json +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/src/alloc/catalog/gpus.v1.json +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/src/alloc/cli.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/src/alloc/code_analyzer.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/src/alloc/config.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/src/alloc/context.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/src/alloc/diagnosis_display.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/src/alloc/diagnosis_engine.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/src/alloc/diagnosis_rules.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/src/alloc/display.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/src/alloc/extractor_runner.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/src/alloc/ghost.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/src/alloc/model_extractor.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/src/alloc/model_registry.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/src/alloc/stability.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/src/alloc/upload.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/src/alloc/yaml_config.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/src/alloc.egg-info/SOURCES.txt +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/src/alloc.egg-info/dependency_links.txt +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/src/alloc.egg-info/entry_points.txt +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/src/alloc.egg-info/requires.txt +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/src/alloc.egg-info/top_level.txt +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/tests/test_artifact.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/tests/test_artifact_loader.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/tests/test_auth.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/tests/test_callbacks.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/tests/test_catalog.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/tests/test_cli.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/tests/test_code_analyzer.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/tests/test_context.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/tests/test_diagnose_cli.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/tests/test_diagnosis_engine.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/tests/test_diagnosis_rules.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/tests/test_extractor_activation.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/tests/test_ghost.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/tests/test_ghost_degradation.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/tests/test_init_from_org.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/tests/test_interconnect.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/tests/test_model_extractor.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/tests/test_probe_hw.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/tests/test_probe_multi.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/tests/test_scan_auth.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/tests/test_stability.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/tests/test_topology_strategy.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/tests/test_upload.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/tests/test_verdict.py +0 -0
- {alloc-0.0.11 → alloc-0.0.12}/tests/test_yaml_config.py +0 -0
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.4
|
|
2
2
|
Name: alloc
|
|
3
|
-
Version: 0.0.
|
|
3
|
+
Version: 0.0.12
|
|
4
4
|
Summary: Engineer-first training calibration: estimate VRAM fit, profile short runs, and pick GPU configs under real budget constraints.
|
|
5
5
|
Author-email: Alloc Labs <hello@alloclabs.com>
|
|
6
6
|
License-Expression: Apache-2.0
|
|
@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
|
|
|
4
4
|
|
|
5
5
|
[project]
|
|
6
6
|
name = "alloc"
|
|
7
|
-
version = "0.0.
|
|
7
|
+
version = "0.0.12"
|
|
8
8
|
description = "Engineer-first training calibration: estimate VRAM fit, profile short runs, and pick GPU configs under real budget constraints."
|
|
9
9
|
readme = "README.md"
|
|
10
10
|
license = "Apache-2.0"
|
|
@@ -9,7 +9,7 @@ _warnings.filterwarnings("ignore", category=FutureWarning, module=r"torch\.cuda"
|
|
|
9
9
|
_warnings.filterwarnings("ignore", category=DeprecationWarning, module=r"torch\.cuda")
|
|
10
10
|
del _warnings
|
|
11
11
|
|
|
12
|
-
__version__ = "0.0.
|
|
12
|
+
__version__ = "0.0.12"
|
|
13
13
|
|
|
14
14
|
from alloc.ghost import ghost, GhostReport
|
|
15
15
|
from alloc.callbacks import AllocCallback as HuggingFaceCallback
|
|
@@ -510,6 +510,16 @@ def probe_command(
|
|
|
510
510
|
pmap.append({"gpu_index": idx})
|
|
511
511
|
num_gpus_ref[0] = len(handles)
|
|
512
512
|
process_map_ref[0] = pmap
|
|
513
|
+
# Immediately sample all discovered GPUs so per_gpu_peaks
|
|
514
|
+
# is populated even if the process exits right after discovery
|
|
515
|
+
pgp = per_gpu_peaks_ref[0]
|
|
516
|
+
for gi, h in enumerate(handles):
|
|
517
|
+
try:
|
|
518
|
+
mi = pynvml.nvmlDeviceGetMemoryInfo(h)
|
|
519
|
+
vm = mi.used / (1024 * 1024)
|
|
520
|
+
pgp[gi] = max(pgp.get(gi, 0.0), vm)
|
|
521
|
+
except Exception:
|
|
522
|
+
pass
|
|
513
523
|
except Exception:
|
|
514
524
|
pass
|
|
515
525
|
# Detect interconnect type between discovered GPUs
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.4
|
|
2
2
|
Name: alloc
|
|
3
|
-
Version: 0.0.
|
|
3
|
+
Version: 0.0.12
|
|
4
4
|
Summary: Engineer-first training calibration: estimate VRAM fit, profile short runs, and pick GPU configs under real budget constraints.
|
|
5
5
|
Author-email: Alloc Labs <hello@alloclabs.com>
|
|
6
6
|
License-Expression: Apache-2.0
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|