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()