Skip to content

GitLab

"git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "b089102a8ee42117101a34adc62a0069129262be"
Unverified Commit a90b3978 authored by Miao Zheng's avatar Miao Zheng Committed by GitHub
Browse files

[Fix] Unrecognized symbol in correlation.cpp (#1372)

parent 745aa737
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
mmcv/ops/csrc/pytorch/correlation.cpp
View file @ a90b3978
...@@ -48,7 +48,7 @@ void correlation_forward(Tensor input1, Tensor input2, Tensor output, int kH, ...@@ -48,7 +48,7 @@ void correlation_forward(Tensor input1, Tensor input2, Tensor output, int kH,
int kW, int patchH, int patchW, int padH, int padW, int kW, int patchH, int patchW, int padH, int padW,
int dilationH, int dilationW, int dilation_patchH, int dilationH, int dilationW, int dilation_patchH,
int dilation_patchW, int dH, int dW) { int dilation_patchW, int dH, int dW) {
if (input1.device().is_cuda() and input2.device().is_cuda()) { if (input1.device().is_cuda() && input2.device().is_cuda()) {
#ifdef MMCV_WITH_CUDA #ifdef MMCV_WITH_CUDA
CHECK_CUDA_INPUT(input1); CHECK_CUDA_INPUT(input1);
CHECK_CUDA_INPUT(input2); CHECK_CUDA_INPUT(input2);
...@@ -68,7 +68,7 @@ void correlation_backward(Tensor grad_output, Tensor input1, Tensor input2, ...@@ -68,7 +68,7 @@ void correlation_backward(Tensor grad_output, Tensor input1, Tensor input2,
int kW, int patchH, int patchW, int padH, int padW, int kW, int patchH, int patchW, int padH, int padW,
int dilationH, int dilationW, int dilation_patchH, int dilationH, int dilationW, int dilation_patchH,
int dilation_patchW, int dH, int dW) { int dilation_patchW, int dH, int dW) {
if (input1.device().is_cuda() and input2.device().is_cuda()) { if (input1.device().is_cuda() && input2.device().is_cuda()) {
#ifdef MMCV_WITH_CUDA #ifdef MMCV_WITH_CUDA
CHECK_CUDA_INPUT(grad_output); CHECK_CUDA_INPUT(grad_output);
CHECK_CUDA_INPUT(input1); CHECK_CUDA_INPUT(input1);
......
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