kernelmeter 0.4.1__tar.gz → 0.4.2__tar.gz

{kernelmeter-0.4.1/src/kernelmeter.egg-info → kernelmeter-0.4.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kernelmeter
-Version: 0.4.1
+Version: 0.4.2
 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.4.1 → kernelmeter-0.4.2}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "kernelmeter"
-version = "0.4.1"
+version = "0.4.2"
 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.4.1 → kernelmeter-0.4.2}/src/kernelmeter/__init__.py

@@ -8,7 +8,7 @@ from .extras import DeviceExtras
 from .occupancy import Occupancy
 from .peaks import Peaks
 
-__version__ = "0.4.1"
+__version__ = "0.4.2"
 
 __all__ = [
     "BenchResult",

{kernelmeter-0.4.1 → kernelmeter-0.4.2}/src/kernelmeter/nvml.py

@@ -35,7 +35,9 @@ ARCH_NAMES = {
     10: "Blackwell",
 }
 
-# nvmlBrandType_t (common entries)
+# nvmlBrandType_t. Modern data-center cards report NVML_BRAND_NVIDIA (14)
+# rather than the older TESLA (2), so the 12-16 entries matter on recent
+# drivers -- without them a current Tesla/A/H-series card comes back blank.
 BRAND_NAMES = {
     0: "Unknown",
     1: "Quadro",
@@ -49,6 +51,11 @@ BRAND_NAMES = {
     9: "NVIDIA vCS",
     10: "NVIDIA vWS",
     11: "NVIDIA Cloud Gaming",
+    12: "Quadro RTX",
+    13: "NVIDIA RTX",
+    14: "NVIDIA",
+    15: "GeForce RTX",
+    16: "Titan RTX",
 }
 
 

{kernelmeter-0.4.1 → kernelmeter-0.4.2/src/kernelmeter.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: kernelmeter
-Version: 0.4.1
+Version: 0.4.2
 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.4.1 → kernelmeter-0.4.2}/tests/test_extras.py

@@ -10,7 +10,7 @@ def _fake_nvml():
 def test_from_nvml_builds_extras():
     ex = extras.from_nvml(_fake_nvml(), 0)
     assert ex.architecture == "Turing"
-    assert ex.brand == "Tesla"
+    assert ex.brand == "NVIDIA"
     assert ex.num_gpu_cores == 2560
     assert ex.memory_total_bytes == 15843721216
     assert ex.pcie_gen_current == 3

{kernelmeter-0.4.1 → kernelmeter-0.4.2}/tests/test_nvml.py

@@ -51,7 +51,7 @@ class FakeNvmlLib:
         return NVML_SUCCESS
 
     def nvmlDeviceGetBrand(self, handle, ptr):
-        ptr._obj.value = 2  # Tesla
+        ptr._obj.value = 14  # NVML_BRAND_NVIDIA, what modern T4 drivers report
         return NVML_SUCCESS
 
     def nvmlDeviceGetNumGpuCores(self, handle, ptr):
@@ -119,7 +119,7 @@ def test_static_device_facts():
     n = nvml.Nvml(lib=FakeNvmlLib())
     h = n.device(0)
     assert nvml.ARCH_NAMES[n.architecture(h)] == "Turing"
-    assert nvml.BRAND_NAMES[n.brand(h)] == "Tesla"
+    assert nvml.BRAND_NAMES[n.brand(h)] == "NVIDIA"  # value 14, the modern range
     assert n.num_gpu_cores(h) == 2560
     total, free, used = n.memory_info(h)
     assert total == 15843721216