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
FAST-RNNT
Commits
89beacff
Commit
89beacff
authored
Jul 02, 2021
by
Daniel Povey
Browse files
Remove debug statements
parent
b3c1340d
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
50 additions
and
44 deletions
+50
-44
README.md
README.md
+46
-44
torch_integrated_conv/integrated_conv_cuda_kernel.cu
torch_integrated_conv/integrated_conv_cuda_kernel.cu
+4
-0
No files found.
README.md
View file @
89beacff
# TODO: update this README!
# Fast Discounted Cumulative Sums in PyTorch
[

](https://pypi.org/project/torch-discounted-cumsum/)
...
...
torch_integrated_conv/integrated_conv_cuda_kernel.cu
View file @
89beacff
...
...
@@ -665,6 +665,7 @@ torch::Tensor integrated_conv_cuda(torch::Tensor input,
assert
(
num_blocks_patch
<=
num_patches
&&
num_blocks_batch
<=
N
);
#if 0
std::cout << "N,C,H,W=" << N << "," << C << "," << H << "," << W
<< "; kW,kH=" << kW << "," << kH
<< "; patchH,patchW=" << patchH << ","
...
...
@@ -674,6 +675,7 @@ torch::Tensor integrated_conv_cuda(torch::Tensor input,
<< ", threads_per_opixel=" << threads_per_opixel
<< ", threads_per_block=" << threads_per_block
<< std::endl;
#endif
dim3
gridDim
(
C
,
num_blocks_patch
,
num_blocks_batch
);
// blockDim is scalar, just threads_per_block.
...
...
@@ -805,6 +807,7 @@ std::vector<torch::Tensor> integrated_conv_backward_cuda(torch::Tensor input,
assert
(
patchH
*
patchW
*
threads_per_pixel
<=
threads_per_block
);
assert
(
kH
*
kW
*
threads_per_kernel_pos
<=
threads_per_block
);
#if 0
std::cout << "[backward:] N,C,H,W=" << N << "," << C << "," << H << "," << W
<< "; kW,kH=" << kW << "," << kH
<< "; patchH,patchW=" << patchH << ","
...
...
@@ -816,6 +819,7 @@ std::vector<torch::Tensor> integrated_conv_backward_cuda(torch::Tensor input,
<< ", threads_per_block=" << threads_per_block
<< ", buffer_numel=" << buffer_numel
<< std::endl;
#endif
int
num_blocks
=
num_blocks_patch
*
num_blocks_batch
;
...
...
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