mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 00:21:07 +01:00
64b8d20a5c
This PR fixes typos in comments and messages of files under `c10` directory Pull Request resolved: https://github.com/pytorch/pytorch/pull/98079 Approved by: https://github.com/Skylion007
30 lines
667 B
C++
30 lines
667 B
C++
#pragma once
|
|
#include <c10/macros/Macros.h>
|
|
|
|
// Utility to guarantee complete unrolling of a loop where the bounds are known
|
|
// at compile time. Various pragmas achieve similar effects, but are not as
|
|
// portable across compilers.
|
|
|
|
// Example: c10::ForcedUnroll<4>{}(f); is equivalent to f(0); f(1); f(2); f(3);
|
|
|
|
namespace c10 {
|
|
|
|
template <int n>
|
|
struct ForcedUnroll {
|
|
template <typename Func>
|
|
C10_ALWAYS_INLINE void operator()(const Func& f) const {
|
|
ForcedUnroll<n - 1>{}(f);
|
|
f(n - 1);
|
|
}
|
|
};
|
|
|
|
template <>
|
|
struct ForcedUnroll<1> {
|
|
template <typename Func>
|
|
C10_ALWAYS_INLINE void operator()(const Func& f) const {
|
|
f(0);
|
|
}
|
|
};
|
|
|
|
} // namespace c10
|