From 6b6a0fe3d858f56e757e6da44a18b48631852ad8 Mon Sep 17 00:00:00 2001 From: Joseph Magly <1159087+jmagly@users.noreply.github.com> Date: Sun, 23 Aug 2026 11:46:39 -0400 Subject: [PATCH] test(device): restore release branch coverage --- obliteratus/__init__.py | 2 +- pyproject.toml | 2 +- tests/test_device_boundaries.py | 14 ++++++++++++++ uv.lock | 2 +- 4 files changed, 17 insertions(+), 3 deletions(-) diff --git a/obliteratus/__init__.py b/obliteratus/__init__.py index 0df4f55..7707d30 100644 --- a/obliteratus/__init__.py +++ b/obliteratus/__init__.py @@ -1,6 +1,6 @@ """Obliteratus — Master Ablation Suite for HuggingFace transformers.""" -__version__ = "0.1.2" +__version__ = "0.1.3" # Lazy imports for the main pipeline classes __all__ = [ diff --git a/pyproject.toml b/pyproject.toml index 1bd7a82..3cc3634 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta" [project] name = "obliteratus" -version = "0.1.2" +version = "0.1.3" description = "Master Ablation Suite for HuggingFace transformers" readme = "README.md" requires-python = ">=3.10" diff --git a/tests/test_device_boundaries.py b/tests/test_device_boundaries.py index 13b8625..bc60004 100644 --- a/tests/test_device_boundaries.py +++ b/tests/test_device_boundaries.py @@ -248,3 +248,17 @@ def test_configure_cuda_can_disable_cudnn(monkeypatch): disable.assert_called_once_with(False) assert device.torch.backends.cudnn.enabled is False + + +def test_configure_cuda_disables_cudnn_without_sdp_toggle(monkeypatch): + """Older Torch builds may not expose the cuDNN SDP backend toggle.""" + monkeypatch.setattr(device, "is_cuda", lambda: True) + monkeypatch.delattr( + device.torch.backends.cuda, "enable_cudnn_sdp", raising=False + ) + monkeypatch.setenv("OBLITERATUS_DISABLE_CUDNN", "1") + monkeypatch.setattr(device.torch.backends.cudnn, "enabled", True) + + device.configure_cuda_alloc() + + assert device.torch.backends.cudnn.enabled is False diff --git a/uv.lock b/uv.lock index bb2c903..b0f32b7 100644 --- a/uv.lock +++ b/uv.lock @@ -2598,7 +2598,7 @@ wheels = [ [[package]] name = "obliteratus" -version = "0.1.2" +version = "0.1.3" source = { editable = "." } dependencies = [ { name = "accelerate" },