mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
6fece41e9a
Summary: `-Wextra-semi` or `-Wextra-semi-stmt` If the code compiles, this is safe to land. Test Plan: Sandcastle Reviewed By: palmje Differential Revision: D51995078 Pull Request resolved: https://github.com/pytorch/pytorch/pull/115761 Approved by: https://github.com/Skylion007
36 lines
659 B
C++
36 lines
659 B
C++
#pragma once
|
|
|
|
// See Note [hip-clang differences to hcc]
|
|
|
|
#if defined(__CUDA_ARCH__) || defined(__HIP_DEVICE_COMPILE__) || \
|
|
defined(__HIP__) || (defined(__clang__) && defined(__CUDA__))
|
|
#define CONVERSIONS_DECL __host__ __device__ inline
|
|
#else
|
|
#define CONVERSIONS_DECL inline
|
|
#endif
|
|
|
|
#ifdef _MSC_VER
|
|
#undef IN
|
|
#undef OUT
|
|
#endif
|
|
|
|
namespace caffe2 {
|
|
|
|
namespace convert {
|
|
|
|
template <typename IN, typename OUT>
|
|
CONVERSIONS_DECL OUT To(const IN in) {
|
|
return static_cast<OUT>(in);
|
|
}
|
|
|
|
template <typename OUT, typename IN>
|
|
CONVERSIONS_DECL OUT Get(IN x) {
|
|
return static_cast<OUT>(x);
|
|
}
|
|
|
|
} // namespace convert
|
|
|
|
} // namespace caffe2
|
|
|
|
#undef CONVERSIONS_DECL
|