mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
700941f683
Clang-tidy was not applied properly to headers in c10 as documented #91406. These are the easy automated fixes that came out of applying clang-tidy to the c10 part of the code base. cc @ezyang Pull Request resolved: https://github.com/pytorch/pytorch/pull/91407 Approved by: https://github.com/ezyang
71 lines
1.9 KiB
C++
71 lines
1.9 KiB
C++
#pragma once
|
|
|
|
#include <c10/core/SymInt.h>
|
|
#include <c10/util/ArrayRef.h>
|
|
#include <c10/util/Exception.h>
|
|
#include <c10/util/Optional.h>
|
|
|
|
#include <array>
|
|
#include <initializer_list>
|
|
#include <iterator>
|
|
#include <vector>
|
|
|
|
namespace c10 {
|
|
using SymIntArrayRef = ArrayRef<SymInt>;
|
|
|
|
inline at::IntArrayRef asIntArrayRefUnchecked(c10::SymIntArrayRef ar) {
|
|
return IntArrayRef(reinterpret_cast<const int64_t*>(ar.data()), ar.size());
|
|
}
|
|
|
|
inline c10::optional<at::IntArrayRef> asIntArrayRefSlowOpt(
|
|
c10::SymIntArrayRef ar) {
|
|
for (const c10::SymInt& sci : ar) {
|
|
if (sci.is_symbolic()) {
|
|
return c10::nullopt;
|
|
}
|
|
}
|
|
|
|
return {asIntArrayRefUnchecked(ar)};
|
|
}
|
|
|
|
inline at::IntArrayRef asIntArrayRefSlow(
|
|
c10::SymIntArrayRef ar,
|
|
const char* file,
|
|
int64_t line) {
|
|
for (const c10::SymInt& sci : ar) {
|
|
TORCH_CHECK(
|
|
!sci.is_symbolic(),
|
|
file,
|
|
":",
|
|
line,
|
|
": SymIntArrayRef expected to contain only concrete integers");
|
|
}
|
|
return asIntArrayRefUnchecked(ar);
|
|
}
|
|
|
|
#define C10_AS_INTARRAYREF_SLOW(a) c10::asIntArrayRefSlow(a, __FILE__, __LINE__)
|
|
|
|
// Prefer using a more semantic constructor, like
|
|
// fromIntArrayRefKnownNonNegative
|
|
inline SymIntArrayRef fromIntArrayRefUnchecked(IntArrayRef array_ref) {
|
|
return SymIntArrayRef(
|
|
reinterpret_cast<const SymInt*>(array_ref.data()), array_ref.size());
|
|
}
|
|
|
|
inline SymIntArrayRef fromIntArrayRefKnownNonNegative(IntArrayRef array_ref) {
|
|
return fromIntArrayRefUnchecked(array_ref);
|
|
}
|
|
|
|
inline SymIntArrayRef fromIntArrayRefSlow(IntArrayRef array_ref) {
|
|
for (size_t i = 0; i < array_ref.size(); ++i) {
|
|
TORCH_CHECK(
|
|
SymInt::check_range(array_ref[i]),
|
|
"IntArrayRef contains an int that cannot be represented as a SymInt: ",
|
|
array_ref[i]);
|
|
}
|
|
return SymIntArrayRef(
|
|
reinterpret_cast<const SymInt*>(array_ref.data()), array_ref.size());
|
|
}
|
|
|
|
} // namespace c10
|