Commit 0981ae87 authored by pengcheng888's avatar pengcheng888
Browse files

issue/333- fix compile error.

parent 66dfb435
......@@ -172,7 +172,7 @@ private:
CHECK_CUDNN(cudnnCreateTensorDescriptor(&b_desc));
CHECK_CUDNN(cudnnSetTensorNdDescriptor(
b_desc, cudnn_data_type, bias_dims_arr.size(),
b_desc, cudnn_data_type, static_cast<int>(bias_dims_arr.size()),
bias_dims_arr.data(), bias_strides_arr.data()));
CHECK_CUDNN(cudnnCreateActivationDescriptor(&act_desc));
CHECK_CUDNN(cudnnSetActivationDescriptor(
......
......@@ -20,7 +20,13 @@ target("infiniop-nvidia")
import("lib.detect.find_tool")
local nvcc = find_tool("nvcc")
if nvcc ~= nil then
target:add("linkdirs", path.directory(path.directory(nvcc.program)) .. "/lib64/stubs")
if is_plat("windows") then
nvcc_path = os.iorun("where nvcc"):match("(.-)\r?\n")
else
nvcc_path = nvcc.program
end
target:add("linkdirs", path.directory(path.directory(nvcc_path)) .. "/lib64/stubs")
target:add("links", "cuda")
end
end)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment