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
OpenDAS
torch-spline-conv
Commits
00ec0037
Commit
00ec0037
authored
Mar 14, 2018
by
rusty1s
Browse files
removed tests, added cuda boilerplate
parent
7452b08a
Changes
9
Show whitespace changes
Inline
Side-by-side
Showing
9 changed files
with
39 additions
and
10 deletions
+39
-10
torch_spline_conv/old/edgewise_spline_weighting_gpu.py
torch_spline_conv/old/edgewise_spline_weighting_gpu.py
+0
-0
torch_spline_conv/old/spline_conv2.py
torch_spline_conv/old/spline_conv2.py
+0
-0
torch_spline_conv/old/spline_cubic_gpu.py
torch_spline_conv/old/spline_cubic_gpu.py
+0
-0
torch_spline_conv/old/spline_linear_gpu.py
torch_spline_conv/old/spline_linear_gpu.py
+0
-0
torch_spline_conv/old/spline_quadratic_gpu.py
torch_spline_conv/old/spline_quadratic_gpu.py
+0
-0
torch_spline_conv/src/cpu.h
torch_spline_conv/src/cpu.h
+10
-10
torch_spline_conv/src/cuda.c
torch_spline_conv/src/cuda.c
+9
-0
torch_spline_conv/src/cuda.h
torch_spline_conv/src/cuda.h
+6
-0
torch_spline_conv/src/generic/cuda.c
torch_spline_conv/src/generic/cuda.c
+14
-0
No files found.
torch_spline_conv/old/edgewise_spline_weighting_gpu
_test
.py
→
torch_spline_conv/old/edgewise_spline_weighting_gpu.py
View file @
00ec0037
File moved
torch_spline_conv/old/spline_conv
_test
.py
→
torch_spline_conv/old/spline_conv
2
.py
View file @
00ec0037
File moved
torch_spline_conv/old/spline_cubic_gpu
_test
.py
→
torch_spline_conv/old/spline_cubic_gpu.py
View file @
00ec0037
File moved
torch_spline_conv/old/spline_linear_gpu
_test
.py
→
torch_spline_conv/old/spline_linear_gpu.py
View file @
00ec0037
File moved
torch_spline_conv/old/spline_quadratic_gpu
_test
.py
→
torch_spline_conv/old/spline_quadratic_gpu.py
View file @
00ec0037
File moved
torch_spline_conv/src/cpu.h
View file @
00ec0037
void
spline_linear_basis_forward_Float
(
THFloatTensor
*
basis
,
THLongTensor
*
weight_index
,
THFloatTensor
*
pseudo
,
THLongTensor
*
kernel_size
,
THByteTensor
*
is_open_spline
,
int
K
);
void
spline_linear_basis_forward_Float
(
THFloatTensor
*
basis
,
THLongTensor
*
weight_index
,
THFloatTensor
*
pseudo
,
THLongTensor
*
kernel_size
,
THByteTensor
*
is_open_spline
,
int
K
);
void
spline_linear_basis_forward_Double
(
THDoubleTensor
*
basis
,
THLongTensor
*
weight_index
,
THDoubleTensor
*
pseudo
,
THLongTensor
*
kernel_size
,
THByteTensor
*
is_open_spline
,
int
K
);
void
spline_quadratic_basis_forward_Float
(
THFloatTensor
*
basis
,
THLongTensor
*
weight_index
,
THFloatTensor
*
pseudo
,
THLongTensor
*
kernel_size
,
THByteTensor
*
is_open_spline
,
int
K
);
void
spline_quadratic_basis_forward_Float
(
THFloatTensor
*
basis
,
THLongTensor
*
weight_index
,
THFloatTensor
*
pseudo
,
THLongTensor
*
kernel_size
,
THByteTensor
*
is_open_spline
,
int
K
);
void
spline_quadratic_basis_forward_Double
(
THDoubleTensor
*
basis
,
THLongTensor
*
weight_index
,
THDoubleTensor
*
pseudo
,
THLongTensor
*
kernel_size
,
THByteTensor
*
is_open_spline
,
int
K
);
void
spline_cubic_basis_forward_Float
(
THFloatTensor
*
basis
,
THLongTensor
*
weight_index
,
THFloatTensor
*
pseudo
,
THLongTensor
*
kernel_size
,
THByteTensor
*
is_open_spline
,
int
K
);
void
spline_cubic_basis_forward_Float
(
THFloatTensor
*
basis
,
THLongTensor
*
weight_index
,
THFloatTensor
*
pseudo
,
THLongTensor
*
kernel_size
,
THByteTensor
*
is_open_spline
,
int
K
);
void
spline_cubic_basis_forward_Double
(
THDoubleTensor
*
basis
,
THLongTensor
*
weight_index
,
THDoubleTensor
*
pseudo
,
THLongTensor
*
kernel_size
,
THByteTensor
*
is_open_spline
,
int
K
);
void
spline_linear_basis_backward_Float
(
THFloatTensor
*
grad_pseudo
,
THFloatTensor
*
grad_basis
,
THFloatTensor
*
pseudo
,
THLongTensor
*
kernel_size
,
THByteTensor
*
is_open_spline
);
void
spline_linear_basis_backward_Float
(
THFloatTensor
*
grad_pseudo
,
THFloatTensor
*
grad_basis
,
THFloatTensor
*
pseudo
,
THLongTensor
*
kernel_size
,
THByteTensor
*
is_open_spline
);
void
spline_linear_basis_backward_Double
(
THDoubleTensor
*
grad_pseudo
,
THDoubleTensor
*
grad_basis
,
THDoubleTensor
*
pseudo
,
THLongTensor
*
kernel_size
,
THByteTensor
*
is_open_spline
);
void
spline_quadratic_basis_backward_Float
(
THFloatTensor
*
grad_pseudo
,
THFloatTensor
*
grad_basis
,
THFloatTensor
*
pseudo
,
THLongTensor
*
kernel_size
,
THByteTensor
*
is_open_spline
);
void
spline_quadratic_basis_backward_Float
(
THFloatTensor
*
grad_pseudo
,
THFloatTensor
*
grad_basis
,
THFloatTensor
*
pseudo
,
THLongTensor
*
kernel_size
,
THByteTensor
*
is_open_spline
);
void
spline_quadratic_basis_backward_Double
(
THDoubleTensor
*
grad_pseudo
,
THDoubleTensor
*
grad_basis
,
THDoubleTensor
*
pseudo
,
THLongTensor
*
kernel_size
,
THByteTensor
*
is_open_spline
);
void
spline_cubic_basis_backward_Float
(
THFloatTensor
*
grad_pseudo
,
THFloatTensor
*
grad_basis
,
THFloatTensor
*
pseudo
,
THLongTensor
*
kernel_size
,
THByteTensor
*
is_open_spline
);
void
spline_cubic_basis_backward_Float
(
THFloatTensor
*
grad_pseudo
,
THFloatTensor
*
grad_basis
,
THFloatTensor
*
pseudo
,
THLongTensor
*
kernel_size
,
THByteTensor
*
is_open_spline
);
void
spline_cubic_basis_backward_Double
(
THDoubleTensor
*
grad_pseudo
,
THDoubleTensor
*
grad_basis
,
THDoubleTensor
*
pseudo
,
THLongTensor
*
kernel_size
,
THByteTensor
*
is_open_spline
);
void
spline_weighting_forward_Float
(
THFloatTensor
*
output
,
THFloatTensor
*
input
,
THFloatTensor
*
weight
,
THFloatTensor
*
basis
,
THLongTensor
*
weight_index
);
void
spline_weighting_forward_Float
(
THFloatTensor
*
output
,
THFloatTensor
*
input
,
THFloatTensor
*
weight
,
THFloatTensor
*
basis
,
THLongTensor
*
weight_index
);
void
spline_weighting_forward_Double
(
THDoubleTensor
*
output
,
THDoubleTensor
*
input
,
THDoubleTensor
*
weight
,
THDoubleTensor
*
basis
,
THLongTensor
*
weight_index
);
void
spline_weighting_backward_input_Float
(
THFloatTensor
*
grad_input
,
THFloatTensor
*
grad_output
,
THFloatTensor
*
weight
,
THFloatTensor
*
basis
,
THLongTensor
*
weight_index
);
void
spline_weighting_backward_input_Float
(
THFloatTensor
*
grad_input
,
THFloatTensor
*
grad_output
,
THFloatTensor
*
weight
,
THFloatTensor
*
basis
,
THLongTensor
*
weight_index
);
void
spline_weighting_backward_input_Double
(
THDoubleTensor
*
grad_input
,
THDoubleTensor
*
grad_output
,
THDoubleTensor
*
weight
,
THDoubleTensor
*
basis
,
THLongTensor
*
weight_index
);
void
spline_weighting_backward_basis_Float
(
THFloatTensor
*
grad_basis
,
THFloatTensor
*
grad_output
,
THFloatTensor
*
input
,
THFloatTensor
*
weight
,
THLongTensor
*
weight_index
);
void
spline_weighting_backward_basis_Float
(
THFloatTensor
*
grad_basis
,
THFloatTensor
*
grad_output
,
THFloatTensor
*
input
,
THFloatTensor
*
weight
,
THLongTensor
*
weight_index
);
void
spline_weighting_backward_basis_Double
(
THDoubleTensor
*
grad_basis
,
THDoubleTensor
*
grad_output
,
THDoubleTensor
*
input
,
THDoubleTensor
*
weight
,
THLongTensor
*
weight_index
);
void
spline_weighting_backward_weight_Float
(
THFloatTensor
*
grad_weight
,
THFloatTensor
*
grad_output
,
THFloatTensor
*
input
,
THFloatTensor
*
basis
,
THLongTensor
*
weight_index
);
void
spline_weighting_backward_weight_Float
(
THFloatTensor
*
grad_weight
,
THFloatTensor
*
grad_output
,
THFloatTensor
*
input
,
THFloatTensor
*
basis
,
THLongTensor
*
weight_index
);
void
spline_weighting_backward_weight_Double
(
THDoubleTensor
*
grad_weight
,
THDoubleTensor
*
grad_output
,
THDoubleTensor
*
input
,
THDoubleTensor
*
basis
,
THLongTensor
*
weight_index
);
torch_spline_conv/src/cuda.c
View file @
00ec0037
#include <THC/THC.h>
#define spline_(NAME) TH_CONCAT_4(spline_, NAME, _cuda_, Real)
#define spline_kernel_(NAME) TH_CONCAT_4(spline_, NAME, _kernel_, Real)
#include "generic/cuda.c"
#include "THCGenerateFloatType.h"
#include "generic/cuda.c"
#include "THCGenerateDoubleType.h"
torch_spline_conv/src/cuda.h
View file @
00ec0037
void
spline_linear_basis_forward_cuda_Float
(
THCudaTensor
*
basis
,
THCudaLongTensor
*
weight_index
,
THCudaTensor
*
pseudo
,
THCudaLongTensor
*
kernel_size
,
THCudaByteTensor
*
is_open_spline
,
int
K
);
void
spline_linear_basis_forward_cuda_Double
(
THCudaDoubleTensor
*
basis
,
THCudaLongTensor
*
weight_index
,
THCudaDoubleTensor
*
pseudo
,
THCudaLongTensor
*
kernel_size
,
THCudaByteTensor
*
is_open_spline
,
int
K
);
void
spline_quadratic_basis_forward_cuda_Float
(
THCudaTensor
*
basis
,
THCudaLongTensor
*
weight_index
,
THCudaTensor
*
pseudo
,
THCudaLongTensor
*
kernel_size
,
THCudaByteTensor
*
is_open_spline
,
int
K
);
void
spline_quadratic_basis_forward_cuda_Double
(
THCudaDoubleTensor
*
basis
,
THCudaLongTensor
*
weight_index
,
THCudaDoubleTensor
*
pseudo
,
THCudaLongTensor
*
kernel_size
,
THCudaByteTensor
*
is_open_spline
,
int
K
);
void
spline_cubic_basis_forward_cuda_Float
(
THCudaTensor
*
basis
,
THCudaLongTensor
*
weight_index
,
THCudaTensor
*
pseudo
,
THCudaLongTensor
*
kernel_size
,
THCudaByteTensor
*
is_open_spline
,
int
K
);
void
spline_cubic_basis_forward_cuda_Double
(
THCudaDoubleTensor
*
basis
,
THCudaLongTensor
*
weight_index
,
THCudaDoubleTensor
*
pseudo
,
THCudaLongTensor
*
kernel_size
,
THCudaByteTensor
*
is_open_spline
,
int
K
);
torch_spline_conv/src/generic/cuda.c
View file @
00ec0037
#ifndef THC_GENERIC_FILE
#define THC_GENERIC_FILE "generic/cuda.c"
#else
void
spline_
(
linear_basis_forward
)(
THCTensor
*
basis
,
THCudaLongTensor
*
weight_index
,
THCTensor
*
pseudo
,
THCudaLongTensor
*
kernel_size
,
THCudaByteTensor
*
is_open_spline
,
int
K
)
{
}
void
spline_
(
quadratic_basis_forward
)(
THCTensor
*
basis
,
THCudaLongTensor
*
weight_index
,
THCTensor
*
pseudo
,
THCudaLongTensor
*
kernel_size
,
THCudaByteTensor
*
is_open_spline
,
int
K
)
{
}
void
spline_
(
cubic_basis_forward
)(
THCTensor
*
basis
,
THCudaLongTensor
*
weight_index
,
THCTensor
*
pseudo
,
THCudaLongTensor
*
kernel_size
,
THCudaByteTensor
*
is_open_spline
,
int
K
)
{
}
#endif
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