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-harmonics
Commits
418d9576
Unverified
Commit
418d9576
authored
Aug 27, 2024
by
Thorsten Kurth
Committed by
GitHub
Aug 27, 2024
Browse files
removing some redundant code (#44)
parent
78365cb9
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
0 additions
and
2 deletions
+0
-2
torch_harmonics/_disco_convolution.py
torch_harmonics/_disco_convolution.py
+0
-2
No files found.
torch_harmonics/_disco_convolution.py
View file @
418d9576
...
...
@@ -84,10 +84,8 @@ class _DiscoS2TransposeContractionCuda(torch.autograd.Function):
@
custom_bwd
(
device_type
=
"cuda"
)
def
backward
(
ctx
,
grad_output
):
roff_idx
,
ker_idx
,
row_idx
,
col_idx
,
vals
=
ctx
.
saved_tensors
inp_type
=
grad_output
.
dtype
grad_input
=
disco_cuda_extension
.
forward
(
grad_output
.
contiguous
(),
roff_idx
,
ker_idx
,
row_idx
,
col_idx
,
vals
,
ctx
.
kernel_size
,
ctx
.
nlat_in
,
ctx
.
nlon_in
)
grad_input
=
grad_input
.
to
(
dtype
=
inp_type
)
return
grad_input
,
None
,
None
,
None
,
None
,
None
,
None
,
None
,
None
...
...
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