From d4f6e3de4c2eed1230d31c5d7bc74bebc0f16301 Mon Sep 17 00:00:00 2001
From: Seunghoon Lee <lshqqytiger@naver.com>
Date: Wed, 17 Jul 2024 04:46:29 +0900
Subject: [PATCH] fix zluda installer

---
 modules/zluda_installer.py | 17 +++++++++++++++--
 1 file changed, 15 insertions(+), 2 deletions(-)

diff --git a/modules/zluda_installer.py b/modules/zluda_installer.py
index d24d3152f..82eada27e 100644
--- a/modules/zluda_installer.py
+++ b/modules/zluda_installer.py
@@ -5,7 +5,20 @@ import zipfile
 import platform
 import urllib.request
 from typing import Tuple
-from packaging.version import Version
+
+
+class HIPSDKVersion:
+    major: int
+    minor: int
+
+    def __init__(self, version: str):
+        self.major, self.minor = [int(v) for v in version.strip().split(".")]
+
+    def __gt__(self, other):
+        return self.major * 10 + other.minor > other.major * 10 + other.minor
+
+    def __str__(self):
+        return f"{self.major}.{self.minor}"
 
 
 class HIPSDK:
@@ -27,7 +40,7 @@ class HIPSDK:
             for s in versions:
                 version = None
                 try:
-                    version = Version(s)
+                    version = HIPSDKVersion(s)
                 except Exception:
                     continue
                 if default_version is None: