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
MMCV
Commits
998e4597
Unverified
Commit
998e4597
authored
Feb 18, 2023
by
q.yao
Committed by
GitHub
Feb 18, 2023
Browse files
[Fix] Call operator __half explicit (#2608)
parent
a019d2a3
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
1 addition
and
1 deletion
+1
-1
mmcv/ops/csrc/common/cuda/carafe_cuda_kernel.cuh
mmcv/ops/csrc/common/cuda/carafe_cuda_kernel.cuh
+1
-1
No files found.
mmcv/ops/csrc/common/cuda/carafe_cuda_kernel.cuh
View file @
998e4597
...
@@ -59,7 +59,7 @@ __device__ __forceinline__ phalf warpReduceSum(phalf val) {
...
@@ -59,7 +59,7 @@ __device__ __forceinline__ phalf warpReduceSum(phalf val) {
__PHALF
(
val
)
+=
__shfl_down
(
val
,
offset
);
__PHALF
(
val
)
+=
__shfl_down
(
val
,
offset
);
#else
#else
__PHALF
(
val
)
+=
__PHALF
(
val
)
+=
__shfl_down_sync
(
FULL_MASK
,
static_cast
<
__half
>
(
__PHALF
(
val
)
),
offset
);
__shfl_down_sync
(
FULL_MASK
,
__PHALF
(
val
).
operator
__half
(
),
offset
);
#endif
#endif
return
val
;
return
val
;
}
}
...
...
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