Skip to content

GitLab

Commit f07e60cb authored by Jan Eric Lenssen's avatar Jan Eric Lenssen
Browse files

bugfix degree=3

parent 4cb72f68
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
spline_conv_gpu.py
View file @ f07e60cb
...@@ -281,7 +281,7 @@ __global__ void spline_kernel( ...@@ -281,7 +281,7 @@ __global__ void spline_kernel(
const ${Dtype}* input, ${Dtype}* amount, long* index, const ${Dtype}* input, ${Dtype}* amount, long* index,
const long* kernel_size, const long* is_open_spline, int num_threads) { const long* kernel_size, const long* is_open_spline, int num_threads) {
CUDA_KERNEL_LOOP(idx, num_threads}) { CUDA_KERNEL_LOOP(idx, num_threads) {
const int e_idx = idx / ${k_max}; const int e_idx = idx / ${k_max};
int k_idx = idx % ${k_max}; int k_idx = idx % ${k_max};
......
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