File tree Expand file tree Collapse file tree 2 files changed +16
-10
lines changed Expand file tree Collapse file tree 2 files changed +16
-10
lines changed Original file line number Diff line number Diff line change @@ -348,10 +348,11 @@ at::Tensor ROIAlignRotated_forward_cuda(
348348 auto output_size = num_rois * pooled_height * pooled_width * channels;
349349 cudaStream_t stream = at::cuda::getCurrentCUDAStream ();
350350
351- dim3 grid (std::min (
352- at::cuda::ATenCeilDiv (
353- static_cast <int64_t >(output_size), static_cast <int64_t >(512 )),
354- static_cast <int64_t >(4096 )));
351+ dim3 grid (
352+ std::min (
353+ at::cuda::ATenCeilDiv (
354+ static_cast <int64_t >(output_size), static_cast <int64_t >(512 )),
355+ static_cast <int64_t >(4096 )));
355356 dim3 block (512 );
356357
357358 if (output.numel () == 0 ) {
@@ -407,10 +408,11 @@ at::Tensor ROIAlignRotated_backward_cuda(
407408
408409 cudaStream_t stream = at::cuda::getCurrentCUDAStream ();
409410
410- dim3 grid (std::min (
411- at::cuda::ATenCeilDiv (
412- static_cast <int64_t >(grad.numel ()), static_cast <int64_t >(512 )),
413- static_cast <int64_t >(4096 )));
411+ dim3 grid (
412+ std::min (
413+ at::cuda::ATenCeilDiv (
414+ static_cast <int64_t >(grad.numel ()), static_cast <int64_t >(512 )),
415+ static_cast <int64_t >(4096 )));
414416 dim3 block (512 );
415417
416418 // handle possibly empty gradients
Original file line number Diff line number Diff line change @@ -57,7 +57,9 @@ std::string get_compiler_version() {
5757#error "GCC >= 4.9 is required!"
5858#endif
5959
60- { ss << " GCC " << __GNUC__ << " ." << __GNUC_MINOR__; }
60+ {
61+ ss << " GCC " << __GNUC__ << " ." << __GNUC_MINOR__;
62+ }
6163#endif
6264#endif
6365
@@ -69,7 +71,9 @@ std::string get_compiler_version() {
6971#endif
7072
7173#if defined(_MSC_VER)
72- { ss << " MSVC " << _MSC_FULL_VER; }
74+ {
75+ ss << " MSVC " << _MSC_FULL_VER;
76+ }
7377#endif
7478 return ss.str ();
7579}
You can’t perform that action at this time.
0 commit comments