mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
03de807d81
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/60677 Add a rule to wrap conversions.h and depend on that, rather than relying on a glob which violates package boundaries. Test Plan: `buck2 build fbcode//caffe2/caffe2:caffe2_core` Reviewed By: mzlee Differential Revision: D29370841 fbshipit-source-id: d4dd383eb8457d4f5118574e34e6f17c32fde647
36 lines
661 B
C++
36 lines
661 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
|