Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
jerrrrry
infinicore
Commits
0981ae87
Commit
0981ae87
authored
Jul 18, 2025
by
pengcheng888
Browse files
issue/333- fix compile error.
parent
66dfb435
Changes
2
Hide whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
8 additions
and
2 deletions
+8
-2
src/infiniop/ops/conv/nvidia/conv_nvidia.cu
src/infiniop/ops/conv/nvidia/conv_nvidia.cu
+1
-1
xmake/nvidia.lua
xmake/nvidia.lua
+7
-1
No files found.
src/infiniop/ops/conv/nvidia/conv_nvidia.cu
View file @
0981ae87
...
@@ -172,7 +172,7 @@ private:
...
@@ -172,7 +172,7 @@ private:
CHECK_CUDNN
(
cudnnCreateTensorDescriptor
(
&
b_desc
));
CHECK_CUDNN
(
cudnnCreateTensorDescriptor
(
&
b_desc
));
CHECK_CUDNN
(
cudnnSetTensorNdDescriptor
(
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
()));
bias_dims_arr
.
data
(),
bias_strides_arr
.
data
()));
CHECK_CUDNN
(
cudnnCreateActivationDescriptor
(
&
act_desc
));
CHECK_CUDNN
(
cudnnCreateActivationDescriptor
(
&
act_desc
));
CHECK_CUDNN
(
cudnnSetActivationDescriptor
(
CHECK_CUDNN
(
cudnnSetActivationDescriptor
(
...
...
xmake/nvidia.lua
View file @
0981ae87
...
@@ -20,7 +20,13 @@ target("infiniop-nvidia")
...
@@ -20,7 +20,13 @@ target("infiniop-nvidia")
import
(
"lib.detect.find_tool"
)
import
(
"lib.detect.find_tool"
)
local
nvcc
=
find_tool
(
"nvcc"
)
local
nvcc
=
find_tool
(
"nvcc"
)
if
nvcc
~=
nil
then
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"
)
target
:
add
(
"links"
,
"cuda"
)
end
end
end
)
end
)
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment