mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 21:14:14 +08:00
27c5bbe5cb
Pull Request resolved: https://github.com/pytorch/pytorch/pull/119975 Approved by: https://github.com/jbschlosser ghstack dependencies: #119661, #119974
33 lines
1.2 KiB
C++
33 lines
1.2 KiB
C++
#include <c10/core/ConstantSymNodeImpl.h>
|
|
|
|
namespace c10 {
|
|
|
|
// This is used to support the case where the lhs is a constant symnode
|
|
// and the rhs is a nested int symnode. This situation occurs today when we
|
|
// perform a binary op between nested int and plain int and the
|
|
// int is promoted into a constant symnode. If we'd like to
|
|
// support more combinations in the future, we may need to implement some
|
|
// kind of multiple dispatch.
|
|
#define DEFINE_BINARY_OP(OP, ROP) \
|
|
template <typename T> \
|
|
c10::SymNode ConstantSymNodeImpl<T>::OP(const c10::SymNode& other) { \
|
|
TORCH_INTERNAL_ASSERT(other->is_nested_int()); \
|
|
return other->ROP( \
|
|
c10::intrusive_ptr<ConstantSymNodeImpl<T>>::reclaim_copy(this)); \
|
|
}
|
|
|
|
DEFINE_BINARY_OP(eq, eq)
|
|
DEFINE_BINARY_OP(ne, ne)
|
|
DEFINE_BINARY_OP(ge, le)
|
|
DEFINE_BINARY_OP(le, ge)
|
|
DEFINE_BINARY_OP(lt, gt)
|
|
DEFINE_BINARY_OP(gt, lt)
|
|
DEFINE_BINARY_OP(mul, mul)
|
|
|
|
#undef DEFINE_BINARY_OP
|
|
|
|
template class ConstantSymNodeImpl<bool>;
|
|
template class ConstantSymNodeImpl<int64_t>;
|
|
|
|
} // namespace c10
|