@@ -19,59 +19,77 @@ def is_amd64_arch():
1919
2020
2121if sys .platform == "darwin" :
22- if is_amd64_arch ():
23- with open (f"{ LIB_ROOT } /lib/darwin-amd64/libkclvm_cli_cdylib.dylib" , "rb" ) as f :
24- DARWIN_AMD64_CLI_LIB = f .read ()
25- else :
26- with open (f"{ LIB_ROOT } /lib/darwin-arm64/libkclvm_cli_cdylib.dylib" , "rb" ) as f :
27- DARWIN_ARM64_CLI_LIB = f .read ()
2822
2923 def cli_lib ():
24+ if is_amd64_arch ():
25+ with open (
26+ f"{ LIB_ROOT } /lib/darwin-amd64/libkclvm_cli_cdylib.dylib" , "rb"
27+ ) as f :
28+ DARWIN_AMD64_CLI_LIB = f .read ()
29+ else :
30+ with open (
31+ f"{ LIB_ROOT } /lib/darwin-arm64/libkclvm_cli_cdylib.dylib" , "rb"
32+ ) as f :
33+ DARWIN_ARM64_CLI_LIB = f .read ()
3034 return DARWIN_AMD64_CLI_LIB if is_amd64_arch () else DARWIN_ARM64_CLI_LIB
3135
3236 def lib_path () -> str :
3337 if is_amd64_arch ():
3438 return str (LIB_ROOT .joinpath ("lib" ).joinpath ("darwin-amd64" ))
3539 return str (LIB_ROOT .joinpath ("lib" ).joinpath ("darwin-arm64" ))
3640
41+ def lib_name () -> str :
42+ return "libkclvm_cli_cdylib.dylib"
43+
3744elif sys .platform .startswith ("linux" ):
38- if is_amd64_arch ():
39- with open (f"{ LIB_ROOT } /lib/linux-amd64/libkclvm_cli_cdylib.so" , "rb" ) as f :
40- LINUX_AMD64_CLI_LIB = f .read ()
41- else :
42- with open (f"{ LIB_ROOT } /lib/linux-arm64/libkclvm_cli_cdylib.so" , "rb" ) as f :
43- LINUX_ARM64_CLI_LIB = f .read ()
4445
4546 def cli_lib ():
47+ if is_amd64_arch ():
48+ with open (f"{ LIB_ROOT } /lib/linux-amd64/libkclvm_cli_cdylib.so" , "rb" ) as f :
49+ LINUX_AMD64_CLI_LIB = f .read ()
50+ else :
51+ with open (f"{ LIB_ROOT } /lib/linux-arm64/libkclvm_cli_cdylib.so" , "rb" ) as f :
52+ LINUX_ARM64_CLI_LIB = f .read ()
4653 return LINUX_AMD64_CLI_LIB if is_amd64_arch () else LINUX_ARM64_CLI_LIB
4754
4855 def lib_path () -> str :
4956 if is_amd64_arch ():
5057 return str (LIB_ROOT .joinpath ("lib" ).joinpath ("linux-amd64" ))
5158 return str (LIB_ROOT .joinpath ("lib" ).joinpath ("linux-arm64" ))
5259
60+ def lib_name () -> str :
61+ return "libkclvm_cli_cdylib.so"
62+
5363elif sys .platform == "win32" :
54- if is_amd64_arch ():
55- with open (f"{ LIB_ROOT } /lib/windows-amd64/kclvm_cli_cdylib.dll" , "rb" ) as f :
56- WINDOWS_AMD64_CLI_LIB = f .read ()
57- with open (f"{ LIB_ROOT } /lib/windows-amd64/kclvm_cli_cdylib.lib" , "rb" ) as f :
58- WINDOWS_AMD64_EXPORT_LIB = f .read ()
59- else :
60- with open (f"{ LIB_ROOT } /lib/windows-arm64/kclvm_cli_cdylib.dll" , "rb" ) as f :
61- WINDOWS_ARM64_CLI_LIB = f .read ()
62- with open (f"{ LIB_ROOT } /lib/windows-arm64/kclvm_cli_cdylib.lib" , "rb" ) as f :
63- WINDOWS_ARM64_EXPORT_LIB = f .read ()
6464
6565 def cli_lib ():
66+ if is_amd64_arch ():
67+ with open (f"{ LIB_ROOT } /lib/windows-amd64/kclvm_cli_cdylib.dll" , "rb" ) as f :
68+ WINDOWS_AMD64_CLI_LIB = f .read ()
69+ else :
70+ with open (f"{ LIB_ROOT } /lib/windows-arm64/kclvm_cli_cdylib.dll" , "rb" ) as f :
71+ WINDOWS_ARM64_CLI_LIB = f .read ()
6672 return WINDOWS_AMD64_CLI_LIB if is_amd64_arch () else WINDOWS_ARM64_CLI_LIB
6773
6874 def export_lib ():
75+ if is_amd64_arch ():
76+ with open (f"{ LIB_ROOT } /lib/windows-amd64/kclvm_cli_cdylib.lib" , "rb" ) as f :
77+ WINDOWS_AMD64_EXPORT_LIB = f .read ()
78+ else :
79+ with open (f"{ LIB_ROOT } /lib/windows-arm64/kclvm_cli_cdylib.lib" , "rb" ) as f :
80+ WINDOWS_ARM64_EXPORT_LIB = f .read ()
6981 return WINDOWS_AMD64_EXPORT_LIB if is_amd64_arch () else WINDOWS_ARM64_EXPORT_LIB
7082
7183 def lib_path () -> str :
7284 if is_amd64_arch ():
7385 return str (LIB_ROOT .joinpath ("lib" ).joinpath ("windows-amd64" ))
7486 return str (LIB_ROOT .joinpath ("lib" ).joinpath ("windows-arm64" ))
7587
88+ def lib_name () -> str :
89+ return "kclvm_cli_cdylib.dll"
90+
91+ def export_lib_name () -> str :
92+ return "kclvm_cli_cdylib.lib"
93+
7694else :
7795 raise f"Unsupported platform { sys .platform } , expected win32, linux or darwin platform"
0 commit comments