@@ -19,59 +19,77 @@ def is_amd64_arch():
19
19
20
20
21
21
if 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 ()
28
22
29
23
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 ()
30
34
return DARWIN_AMD64_CLI_LIB if is_amd64_arch () else DARWIN_ARM64_CLI_LIB
31
35
32
36
def lib_path () -> str :
33
37
if is_amd64_arch ():
34
38
return str (LIB_ROOT .joinpath ("lib" ).joinpath ("darwin-amd64" ))
35
39
return str (LIB_ROOT .joinpath ("lib" ).joinpath ("darwin-arm64" ))
36
40
41
+ def lib_name () -> str :
42
+ return "libkclvm_cli_cdylib.dylib"
43
+
37
44
elif 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 ()
44
45
45
46
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 ()
46
53
return LINUX_AMD64_CLI_LIB if is_amd64_arch () else LINUX_ARM64_CLI_LIB
47
54
48
55
def lib_path () -> str :
49
56
if is_amd64_arch ():
50
57
return str (LIB_ROOT .joinpath ("lib" ).joinpath ("linux-amd64" ))
51
58
return str (LIB_ROOT .joinpath ("lib" ).joinpath ("linux-arm64" ))
52
59
60
+ def lib_name () -> str :
61
+ return "libkclvm_cli_cdylib.so"
62
+
53
63
elif 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 ()
64
64
65
65
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 ()
66
72
return WINDOWS_AMD64_CLI_LIB if is_amd64_arch () else WINDOWS_ARM64_CLI_LIB
67
73
68
74
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 ()
69
81
return WINDOWS_AMD64_EXPORT_LIB if is_amd64_arch () else WINDOWS_ARM64_EXPORT_LIB
70
82
71
83
def lib_path () -> str :
72
84
if is_amd64_arch ():
73
85
return str (LIB_ROOT .joinpath ("lib" ).joinpath ("windows-amd64" ))
74
86
return str (LIB_ROOT .joinpath ("lib" ).joinpath ("windows-arm64" ))
75
87
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
+
76
94
else :
77
95
raise f"Unsupported platform { sys .platform } , expected win32, linux or darwin platform"
0 commit comments