kernelmeter 0.3.0__tar.gz → 0.3.1__tar.gz

{kernelmeter-0.3.0/src/kernelmeter.egg-info → kernelmeter-0.3.1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kernelmeter
-Version: 0.3.0
+Version: 0.3.1
 Summary: Query every CUDA device attribute without profiling a kernel, and benchmark your kernels against the hardware's speed of light.
 Author: nuemaan
 License: MIT

{kernelmeter-0.3.0 → kernelmeter-0.3.1}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "kernelmeter"
-version = "0.3.0"
+version = "0.3.1"
 description = "Query every CUDA device attribute without profiling a kernel, and benchmark your kernels against the hardware's speed of light."
 readme = "README.md"
 license = { text = "MIT" }

{kernelmeter-0.3.0 → kernelmeter-0.3.1}/src/kernelmeter/__init__.py

@@ -7,7 +7,7 @@ from . import occupancy, roofline
 from .occupancy import Occupancy
 from .peaks import Peaks
 
-__version__ = "0.3.0"
+__version__ = "0.3.1"
 
 __all__ = [
     "BenchResult",

{kernelmeter-0.3.0 → kernelmeter-0.3.1}/src/kernelmeter/attrs.py

@@ -154,6 +154,14 @@ KNOWN_ATTRS: dict[int, str] = {
     141: "host_numa_virtual_memory_management_supported",
     142: "host_numa_memory_pools_supported",
     143: "host_numa_multinode_ipc_supported",
+    144: "host_memory_pools_supported",
+    145: "host_virtual_memory_management_supported",
+    146: "host_alloc_dma_buf_supported",
+    147: "only_partial_host_native_atomic_supported",
+    148: "atomic_reduction_supported",
+    # 149 is CU_DEVICE_ATTRIBUTE_MAX, a sentinel rather than a real
+    # attribute, so it stops here. Anything the driver adds beyond this is
+    # still reported generically as attribute_<id> by the probe below.
 }
 
 

{kernelmeter-0.3.0 → kernelmeter-0.3.1/src/kernelmeter.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kernelmeter
-Version: 0.3.0
+Version: 0.3.1
 Summary: Query every CUDA device attribute without profiling a kernel, and benchmark your kernels against the hardware's speed of light.
 Author: nuemaan
 License: MIT

{kernelmeter-0.3.0 → kernelmeter-0.3.1}/tests/test_attrs.py

@@ -20,8 +20,8 @@ def test_unsupported_ids_are_skipped(fake_driver):
 def test_unknown_but_supported_ids_get_generic_names(fake_driver):
     dev = fake_driver.device(0)
     result = attrs.query_all(fake_driver, dev)
-    # id 150 succeeds in the fake but has no name in our table
-    assert result["attribute_150"] == 7
+    # id 155 succeeds in the fake but has no name in our table
+    assert result["attribute_155"] == 7
 
 
 def test_cuda12_range_names(fake_driver):
@@ -29,6 +29,13 @@ def test_cuda12_range_names(fake_driver):
     result = attrs.query_all(fake_driver, dev)
     assert result["numa_id"] == -1
     assert result["gpu_pci_device_id"] == 0x1EB810DE
+    # last named driver attribute before the CU_DEVICE_ATTRIBUTE_MAX sentinel
+    assert result["atomic_reduction_supported"] == 1
+
+
+def test_max_sentinel_is_not_named():
+    # 149 is CU_DEVICE_ATTRIBUTE_MAX, not a real attribute
+    assert 149 not in attrs.KNOWN_ATTRS
 
 
 def test_device_metadata(fake_driver):