From 5dcd8d242839b247468684a5498f847c44ee5b4c Mon Sep 17 00:00:00 2001
From: Rando717 <respect_killer@outlook.com>
Date: Sun, 21 Sep 2025 19:31:18 +0200
Subject: [PATCH] add files via upload

uploaded nvcuda.zluda_get_nightly_flag.py to get nightly flag info inside batch
---
 .../nvcuda.zluda_get_nightly_flag.py          | 36 +++++++++++++++++++
 1 file changed, 36 insertions(+)
 create mode 100644 comfy/customzluda/nvcuda.zluda_get_nightly_flag.py

diff --git a/comfy/customzluda/nvcuda.zluda_get_nightly_flag.py b/comfy/customzluda/nvcuda.zluda_get_nightly_flag.py
new file mode 100644
index 000000000..cd208203d
--- /dev/null
+++ b/comfy/customzluda/nvcuda.zluda_get_nightly_flag.py
@@ -0,0 +1,36 @@
+import ctypes
+import os
+import sys
+
+def main():
+    dll_path = os.path.abspath(os.path.join(os.path.dirname(__file__), '..', '..', 'zluda', 'nvcuda.dll'))
+
+    if not os.path.isfile(dll_path):
+        print(f"ERROR: DLL not found: {dll_path}")
+        sys.exit(1)
+
+    try:
+        zluda_dll = ctypes.CDLL(dll_path)
+    except Exception as e:
+        print(f"ERROR: Could not load DLL: {e}")
+        sys.exit(1)
+
+    try:
+        zluda_get_nightly_flag = zluda_dll.zluda_get_nightly_flag
+        zluda_get_nightly_flag.restype = ctypes.c_int
+
+        flag = zluda_get_nightly_flag()
+
+        if flag == 1:
+            print("[nightly build]")
+        elif flag == 0:
+            print("[release build]")
+        else:
+            print(f"Unexpected flag value: {flag}")
+
+    except Exception as e:
+        print(f"ERROR: Could not call zluda_get_nightly_flag: {e}")
+        sys.exit(1)
+
+if __name__ == "__main__":
+    main()