Skip to content

Commit f6dfd0a

Browse files
committed
issue/1177: modified nvidia.lua
1 parent c20bba9 commit f6dfd0a

1 file changed

Lines changed: 3 additions & 11 deletions

File tree

xmake/nvidia.lua

Lines changed: 3 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -76,10 +76,6 @@ target("infiniop-nvidia")
7676

7777
if not os.isfile(header_path) then
7878

79-
print("")
80-
print("🔧 First-time generating AWQ Marlin kernels (arch: "
81-
.. tostring(generate_arch) .. ")")
82-
8379
-- save current directory
8480
local oldir = os.curdir()
8581

@@ -103,7 +99,7 @@ target("infiniop-nvidia")
10399
os.cd(oldir)
104100

105101
if not ok then
106-
print("⚠️ generate_kernels.py returned non-zero exit code")
102+
print("generate_kernels.py returned non-zero exit code")
107103
if errors then
108104
print(errors)
109105
end
@@ -112,13 +108,10 @@ target("infiniop-nvidia")
112108

113109
end
114110

115-
print("header_path = " .. header_path)
116-
print("exists = " .. tostring(os.isfile(header_path)))
117-
118111
if os.isfile(header_path) then
119-
print("AWQ Marlin kernels generated successfully!")
112+
print("AWQ Marlin kernels generated successfully!")
120113
else
121-
raise("Failed to generate AWQ Marlin kernels: header missing!")
114+
raise("Failed to generate AWQ Marlin kernels: header missing!")
122115
end
123116
if not arch_opt or type(arch_opt) ~= "string" then
124117
local ok, sm_str = os.iorunv("nvidia-smi", {"--query-gpu=compute_cap", "--format=csv,noheader,nounits"})
@@ -199,7 +192,6 @@ target("infiniop-nvidia")
199192
local sgl_arch = parse_sgl_cuda_arch(arch_opt)
200193
if sgl_arch then
201194
add_defines("SGL_CUDA_ARCH=" .. sgl_arch)
202-
print("SGL_CUDA_ARCH =", sgl_arch)
203195
else
204196
print("Invalid cuda_arch:", arch_opt)
205197
end

0 commit comments

Comments
 (0)