#pragma once #include #include #include #include namespace at { // This is in an extra file to work around strange interaction of // bitset on Windows with operator overloading constexpr size_t dim_bitset_size = 64; static inline std::bitset dim_list_to_bitset(IntArrayRef dims, int64_t ndims) { TORCH_CHECK(ndims <= (int64_t) dim_bitset_size, "only tensors with up to ", dim_bitset_size, " dims are supported"); std::bitset seen; for (size_t i = 0; i < dims.size(); i++) { size_t dim = maybe_wrap_dim(dims[i], ndims); TORCH_CHECK(!seen[dim], "dim ", dim, " appears multiple times in the list of dims"); seen[dim] = true; } return seen; } }