mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
4a019047ad
It is time to enable nested namespaces in the code. Pull Request resolved: https://github.com/pytorch/pytorch/pull/118506 Approved by: https://github.com/albanD
43 lines
1.1 KiB
C++
43 lines
1.1 KiB
C++
#include <c10/core/WrapDimMinimal.h>
|
|
|
|
namespace c10::detail {
|
|
|
|
template <typename T>
|
|
T maybe_wrap_dim_slow(T dim, T dim_post_expr, bool wrap_scalar) {
|
|
TORCH_CHECK_INDEX(
|
|
dim_post_expr >= 0, "Rank cannot be negative but got ", dim_post_expr);
|
|
|
|
if (dim_post_expr == 0) {
|
|
TORCH_CHECK_INDEX(
|
|
wrap_scalar,
|
|
"Dimension specified as ",
|
|
dim,
|
|
" but tensor has no dimensions");
|
|
return c10::maybe_wrap_dim(
|
|
std::move(dim), /*dim_post_expr=*/1, /*wrap_scalar=*/false);
|
|
}
|
|
|
|
T min = dim_post_expr * -1;
|
|
T max = dim_post_expr - 1;
|
|
TORCH_CHECK_INDEX(
|
|
min <= dim && dim <= max,
|
|
"Dimension out of range (expected to be in range of [",
|
|
min,
|
|
", ",
|
|
max,
|
|
"], but got ",
|
|
dim,
|
|
")");
|
|
|
|
TORCH_INTERNAL_ASSERT(
|
|
false, "should never reach here as dim should be out-of-bounds");
|
|
}
|
|
|
|
// Explicitly instantiate the template at the two types it will be used
|
|
template C10_API int64_t
|
|
maybe_wrap_dim_slow(int64_t dim, int64_t dim_post_expr, bool wrap_scalar);
|
|
template C10_API SymInt
|
|
maybe_wrap_dim_slow(SymInt dim, SymInt dim_post_expr, bool wrap_scalar);
|
|
|
|
} // namespace c10::detail
|