gpustack-runtime 0.1.38__py3-none-any.whl → 0.1.38.post2__py3-none-any.whl

gpustack_runtime/_version.py

@@ -27,8 +27,8 @@ version_tuple: VERSION_TUPLE
 __commit_id__: COMMIT_ID
 commit_id: COMMIT_ID
 
-__version__ = version = '0.1.38'
-__version_tuple__ = version_tuple = (0, 1, 38)
+__version__ = version = '0.1.38.post2'
+__version_tuple__ = version_tuple = (0, 1, 38, 'post2')
 try:
     from ._version_appendix import git_commit
     __commit_id__ = commit_id = git_commit

gpustack_runtime/_version_appendix.py

@@ -1 +1 @@
-git_commit = "30f3e75"
+git_commit = "2567d66"

gpustack_runtime/detector/amd.py

@@ -107,8 +107,12 @@ class AMDDetector(Detector):
                     asic_serial = dev_gpu_asic_info.get("asic_serial")
                     dev_uuid = f"GPU-{(asic_serial[2:]).lower()}"
                 else:
-                    dev_uuid = f"GPU-{pyrocmsmi.rsmi_dev_unique_id_get(dev_idx)[2:]}"
-                dev_hsa_agent = hsa_agents.get(dev_uuid)
+                    dev_uuid = ""
+                    with contextlib.suppress(pyrocmsmi.ROCMSMIError):
+                        dev_uuid = (
+                            f"GPU-{pyrocmsmi.rsmi_dev_unique_id_get(dev_idx)[2:]}"
+                        )
+                dev_hsa_agent = hsa_agents.get(dev_uuid, pyhsa.Agent())
 
                 dev_gpu_driver_info = pyamdsmi.amdsmi_get_gpu_driver_info(dev)
                 dev_driver_ver = dev_gpu_driver_info.get("driver_version")
@@ -119,8 +123,13 @@ class AMDDetector(Detector):
 
                 dev_cc = dev_hsa_agent.compute_capability
                 if not dev_cc:
-                    with contextlib.suppress(pyrocmsmi.ROCMSMIError):
-                        dev_cc = pyrocmsmi.rsmi_dev_target_graphics_version_get(dev_idx)
+                    if "target_graphics_version" in dev_gpu_asic_info:
+                        dev_cc = dev_gpu_asic_info.get("target_graphics_version")
+                    else:
+                        with contextlib.suppress(pyrocmsmi.ROCMSMIError):
+                            dev_cc = pyrocmsmi.rsmi_dev_target_graphics_version_get(
+                                dev_idx,
+                            )
 
                 dev_bdf = None
                 dev_card_id = None

gpustack_runtime/detector/hygon.py

@@ -108,7 +108,7 @@ class HygonDetector(Detector):
                 dev_index = dev_idx
 
                 dev_uuid = f"GPU-{pyrocmsmi.rsmi_dev_unique_id_get(dev_idx)[2:]}"
-                dev_hsa_agent = hsa_agents.get(dev_uuid)
+                dev_hsa_agent = hsa_agents.get(dev_uuid, pyhsa.Agent())
 
                 dev_name = dev_hsa_agent.name
                 if not dev_name:

gpustack_runtime/detector/pyhsa/__init__.py

@@ -446,13 +446,13 @@ def has_agent_get_asic_family_id(agent):
 
 @dataclass
 class Agent:
-    device_type: int
-    device_id: str
-    uuid: str
-    name: str
-    compute_capability: str
-    compute_units: int
-    asic_family_id: int | None
+    device_type: int | None = None
+    device_id: str | None = None
+    uuid: str | None = None
+    name: str | None = None
+    compute_capability: str | None = None
+    compute_units: int | None = None
+    asic_family_id: int | None = None
 
 
 def get_agents() -> list[Agent]:

gpustack_runtime/detector/pyrocmsmi/__init__.py

@@ -223,14 +223,7 @@ def rsmi_dev_target_graphics_version_get(device=0):
         c_version = c_uint64()
         ret = rocmsmiLib.rsmi_dev_target_graphics_version_get(device, byref(c_version))
         _rocmsmiCheckReturn(ret)
-        version = str(c_version.value)
-        if len(version) == 4:
-            dev_name = rsmi_dev_name_get(device)
-            if "Instinct MI2" in dev_name:
-                hex_part = str(hex(int(version[2:]))).replace("0x", "")
-                version = version[:2] + hex_part
-        else:
-            version = str(c_version.value // 10 + c_version.value % 10)
+        version = hex(c_version.value)[2:]
         return "gfx" + version
     except AttributeError:
         return None

{gpustack_runtime-0.1.38.dist-info → gpustack_runtime-0.1.38.post2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: gpustack-runtime
-Version: 0.1.38
+Version: 0.1.38.post2
 Summary: GPUStack Runtime is library for detecting GPU resources and launching GPU workloads.
 Project-URL: Homepage, https://github.com/gpustack/runtime
 Project-URL: Bug Tracker, https://github.com/gpustack/gpustack/issues
@@ -15,7 +15,7 @@ Classifier: Programming Language :: Python :: 3.13
 Requires-Python: >=3.10
 Requires-Dist: argcomplete>=3.6.3
 Requires-Dist: docker
-Requires-Dist: gpustack-runner>=0.1.22.post4
+Requires-Dist: gpustack-runner==0.1.22.post6
 Requires-Dist: kubernetes
 Requires-Dist: nvidia-ml-py>=13.580.65
 Requires-Dist: pyyaml

{gpustack_runtime-0.1.38.dist-info → gpustack_runtime-0.1.38.post2.dist-info}/RECORD

@@ -1,6 +1,6 @@
 gpustack_runtime/__init__.py,sha256=kKcK6DMIXOdpWhgMS7xlsNrBNvEmY0L8eyQtBIC3CU4,160
 gpustack_runtime/__main__.py,sha256=Peh1iQnBiQ8GS0lew9rDwlciAHFJn2x7Tm8jyx1PJW8,3488
-gpustack_runtime/_version.py,sha256=YfpTpTWHoaT_k-PZW07GQBpsY-f1kyCVJoFymN_TrlY,777
+gpustack_runtime/_version.py,sha256=ouuYyFNWHR0cA89x0BUEhiLvsudYfbD6bnvl5BAIkko,792
 gpustack_runtime/_version.pyi,sha256=A42NoSgcqEXVy2OeNm4LXC9CbyonbooYrSUBlPm2lGY,156
 gpustack_runtime/envs.py,sha256=qvGn57dVQSdC01QtIQQ7gVRJEls-IvrWXk7WHmPCZM0,25750
 gpustack_runtime/logging.py,sha256=h_fvD5FV7GHCo00IUDLQmkPR-H6r66IX_WSwZwl0mCw,6869
@@ -17,10 +17,10 @@ gpustack_runtime/deployer/kuberentes.py,sha256=Ftvt2cGsrYUf6MZI3czOwGM_G-C5kFLUK
 gpustack_runtime/detector/__init__.py,sha256=kLVxZ9iud2aLwqqAOanIYNb4QSHavUPhqay-FU5ndbU,5443
 gpustack_runtime/detector/__types__.py,sha256=nVh9OV8tZdHFjHPlYul3u2zMvnpa9KOL5GxlPJhEvGk,13163
 gpustack_runtime/detector/__utils__.py,sha256=Yj4GvGvuDuscDG5OpExYdv-1VhmEHBpRrBC9rTsB-kA,24584
-gpustack_runtime/detector/amd.py,sha256=_sRBRZn9pFQ9jITCetvsoHicjXA9RSVrFvo6zOwCHU8,18356
+gpustack_runtime/detector/amd.py,sha256=_9E1168KvMYeax7XDn_oxGmQO2pR7YC9nVuJeV8z1nU,18785
 gpustack_runtime/detector/ascend.py,sha256=VkmNM-enbjMNVXKFfv7z6S-4BkwD6YkBmcC0J1ZGEuc,18588
 gpustack_runtime/detector/cambricon.py,sha256=bqWo2mUuf3a5kV8NC1PzFtHIUhVQkSejUSWR5Z5XaGA,3899
-gpustack_runtime/detector/hygon.py,sha256=I4SWuq5hXLRPJQyIyCoSaU1VnRMDblzQtljKy7CBVdM,13337
+gpustack_runtime/detector/hygon.py,sha256=ieXIPS99ITx8aCyfg7IDoLkE7ZnZNt2SCL2yrmh9uxo,13352
 gpustack_runtime/detector/iluvatar.py,sha256=1fbkfE2w_MH36YgQDv21xb76sl6GvYxVd3qSYXjET0g,10372
 gpustack_runtime/detector/metax.py,sha256=YRpZVfz_nG7QTEMqTxqpgi9x5aPYy4gfvYhiUpbz8R8,10146
 gpustack_runtime/detector/mthreads.py,sha256=PoGirid-rboH4Nm6sKnv424ULax1eu1rEaAsWKPT02g,11105
@@ -30,7 +30,7 @@ gpustack_runtime/detector/pyamdgpu/__init__.py,sha256=vFIOA072L01Nxvzgixns-wNrQJ
 gpustack_runtime/detector/pyamdsmi/__init__.py,sha256=BXAOVnoew6MliU86ci73Rk2yWPeXRUBicFKFWh-r2s0,929
 gpustack_runtime/detector/pycuda/__init__.py,sha256=o2Oj3l0HsXNttwnUylxerbupK-45gZWbluqfEvBDQ38,16673
 gpustack_runtime/detector/pydcmi/__init__.py,sha256=KVmem7Cd6ZoQ-irB_WOsDVOiyGr5Cv69_rm1NNBpkxo,37129
-gpustack_runtime/detector/pyhsa/__init__.py,sha256=jQ6tOPG8kn5Xp3F99-Vu6aUc4hzSUOdoGkEcl8qxQOc,15354
+gpustack_runtime/detector/pyhsa/__init__.py,sha256=v2rWLyyqNQ24Jpy-7MoBbctd797gHN_6JNrFA84AjxM,15445
 gpustack_runtime/detector/pyixml/__init__.py,sha256=oYFAfQ8ZbPg7oMFXFfLjJuCBEV7hWQpFqMMVcN99Sxs,166564
 gpustack_runtime/detector/pymtml/__init__.py,sha256=hfaVZysit_MUTDs0qsnOBgKXq_8DKlCb0Jv00WasU0w,22008
 gpustack_runtime/detector/pymxsml/__init__.py,sha256=spHMs9t2tiV-C9kO-cV_230CBVAZ-j2DWtLfiFU5V3E,25
@@ -38,10 +38,10 @@ gpustack_runtime/detector/pymxsml/mxsml.py,sha256=nRuKSVxIsJj2SJX-SYnE9NGZ7hGjWO
 gpustack_runtime/detector/pymxsml/mxsml_extension.py,sha256=zfWFRGa9eSup336t2lPhIvCbPpa7Prn-7xCHRljg4CI,26967
 gpustack_runtime/detector/pymxsml/mxsml_mcm.py,sha256=a4rX7hJNJKTqLodKU9rYeDaAEKef6UNVThl1w-aiFsA,15363
 gpustack_runtime/detector/pyrocmcore/__init__.py,sha256=8XxKmwLX4-uoP7cfxjGoEmk9qlrGf0804pgyb74mJKs,2108
-gpustack_runtime/detector/pyrocmsmi/__init__.py,sha256=OmvfJepAtEnDKBMLrktMXlRbJEZttMDQ2R4ztD8kLGg,11806
-gpustack_runtime/_version_appendix.py,sha256=1m63CST3al1x5DQ_PR3He7I-46idpBJKuHILVUujnVs,23
-gpustack_runtime-0.1.38.dist-info/METADATA,sha256=R1WSREhvvE1lSRk9sCiI5bptkINlQBnyfWikB-faM8c,1965
-gpustack_runtime-0.1.38.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
-gpustack_runtime-0.1.38.dist-info/entry_points.txt,sha256=bBO_61GxP6dIT74uZwbSDgW5Vt2pTePUS3CgjUJkUgg,68
-gpustack_runtime-0.1.38.dist-info/licenses/LICENSE,sha256=OiPibowBvB-NHV3TP_NOj18XNBlXcshXZFMpa3uvKVE,10362
-gpustack_runtime-0.1.38.dist-info/RECORD,,
+gpustack_runtime/detector/pyrocmsmi/__init__.py,sha256=wuSue_p48-JjkdCK82pd1KS9c42w_xkBPmhGnu5kp-k,11481
+gpustack_runtime/_version_appendix.py,sha256=0uQRTtcOX04gPFN1zUuiipbVLMP5bnddPMs9ZGKfgPU,23
+gpustack_runtime-0.1.38.post2.dist-info/METADATA,sha256=EuFK6z_K-7EfwQh9rfbc-H3NwVX8nPtCmsB7cpYluxc,1971
+gpustack_runtime-0.1.38.post2.dist-info/WHEEL,sha256=WLgqFyCfm_KASv4WHyYy0P3pM_m7J5L9k2skdKLirC8,87
+gpustack_runtime-0.1.38.post2.dist-info/entry_points.txt,sha256=bBO_61GxP6dIT74uZwbSDgW5Vt2pTePUS3CgjUJkUgg,68
+gpustack_runtime-0.1.38.post2.dist-info/licenses/LICENSE,sha256=OiPibowBvB-NHV3TP_NOj18XNBlXcshXZFMpa3uvKVE,10362
+gpustack_runtime-0.1.38.post2.dist-info/RECORD,,