Skip to content

GitLab

"sgl-kernel/python/vscode:/vscode.git/clone" did not exist on "02bc1c7d80799fc04385dc6bcab3dd4c2d7a5f4d"
Unverified Commit 8b539079 authored by Quan (Andy) Gan's avatar Quan (Andy) Gan Committed by GitHub
Browse files

[Bug] Fix #1712 (#1714) 

* fix debug compilation

* fix

* revert
parent 02d31974
Show whitespace changes
Inline Side-by-side
Showing with 47 additions and 0 deletions
src/array/cpu/spmm.h
View file @ 8b539079
......@@ -254,6 +254,8 @@ struct Add {
return *lhs_off + *rhs_off;
}
};
template <typename DType> constexpr bool Add<DType>::use_lhs;
template <typename DType> constexpr bool Add<DType>::use_rhs;
template <typename DType>
struct Sub {
......@@ -263,6 +265,8 @@ struct Sub {
return *lhs_off - *rhs_off;
}
};
template <typename DType> constexpr bool Sub<DType>::use_lhs;
template <typename DType> constexpr bool Sub<DType>::use_rhs;
template <typename DType>
struct Mul {
......@@ -272,6 +276,8 @@ struct Mul {
return *lhs_off * *rhs_off;
}
};
template <typename DType> constexpr bool Mul<DType>::use_lhs;
template <typename DType> constexpr bool Mul<DType>::use_rhs;
template <typename DType>
struct Div {
......@@ -281,6 +287,8 @@ struct Div {
return *lhs_off / *rhs_off;
}
};
template <typename DType> constexpr bool Div<DType>::use_lhs;
template <typename DType> constexpr bool Div<DType>::use_rhs;
template <typename DType>
struct CopyLhs {
......@@ -290,6 +298,8 @@ struct CopyLhs {
return *lhs_off;
}
};
template <typename DType> constexpr bool CopyLhs<DType>::use_lhs;
template <typename DType> constexpr bool CopyLhs<DType>::use_rhs;
template <typename DType>
struct CopyRhs {
......@@ -299,6 +309,8 @@ struct CopyRhs {
return *rhs_off;
}
};
template <typename DType> constexpr bool CopyRhs<DType>::use_lhs;
template <typename DType> constexpr bool CopyRhs<DType>::use_rhs;
//////////////////////////////// Reduce operators on CPU ////////////////////////////////
template <typename DType>
......@@ -309,6 +321,7 @@ struct Max {
return accum < val;
}
};
template <typename DType> constexpr DType Max<DType>::zero;
template <typename DType>
struct Min {
......@@ -318,6 +331,7 @@ struct Min {
return accum > val;
}
};
template <typename DType> constexpr DType Min<DType>::zero;
#define SWITCH_OP(op, Op, ...) \
do { \
......
src/array/cuda/functor.cuh
View file @ 8b539079
......@@ -22,6 +22,9 @@ struct Add {
return lhs[0] + rhs[0];
}
};
template <typename DType> constexpr bool Add<DType>::use_lhs;
template <typename DType> constexpr bool Add<DType>::use_rhs;
template <typename DType> constexpr bool Add<DType>::reduce_last_dim;
template <typename DType>
struct Sub {
......@@ -33,6 +36,9 @@ struct Sub {
return lhs[0] - rhs[0];
}
};
template <typename DType> constexpr bool Sub<DType>::use_lhs;
template <typename DType> constexpr bool Sub<DType>::use_rhs;
template <typename DType> constexpr bool Sub<DType>::reduce_last_dim;
template <typename DType>
struct Mul {
......@@ -44,6 +50,9 @@ struct Mul {
return lhs[0] * rhs[0];
}
};
template <typename DType> constexpr bool Mul<DType>::use_lhs;
template <typename DType> constexpr bool Mul<DType>::use_rhs;
template <typename DType> constexpr bool Mul<DType>::reduce_last_dim;
template <typename DType>
struct Div {
......@@ -55,6 +64,9 @@ struct Div {
return lhs[0] / rhs[0];
}
};
template <typename DType> constexpr bool Div<DType>::use_lhs;
template <typename DType> constexpr bool Div<DType>::use_rhs;
template <typename DType> constexpr bool Div<DType>::reduce_last_dim;
template <typename DType>
struct CopyU {
......@@ -66,6 +78,9 @@ struct CopyU {
return lhs[0];
}
};
template <typename DType> constexpr bool CopyU<DType>::use_lhs;
template <typename DType> constexpr bool CopyU<DType>::use_rhs;
template <typename DType> constexpr bool CopyU<DType>::reduce_last_dim;
template <typename DType>
struct CopyE {
......@@ -77,6 +92,9 @@ struct CopyE {
return rhs[0];
}
};
template <typename DType> constexpr bool CopyE<DType>::use_lhs;
template <typename DType> constexpr bool CopyE<DType>::use_rhs;
template <typename DType> constexpr bool CopyE<DType>::reduce_last_dim;
template <typename DType>
struct Dot {
......@@ -92,6 +110,9 @@ struct Dot {
return rst;
}
};
template <typename DType> constexpr bool Dot<DType>::use_lhs;
template <typename DType> constexpr bool Dot<DType>::use_rhs;
template <typename DType> constexpr bool Dot<DType>::reduce_last_dim;
} // end of namespace binary
......@@ -116,6 +137,10 @@ struct Sum {
Idx *arg_u_buf, Idx *arg_e_buf,
DType val, DType val_ref, Idx uid, Idx eid) {}
};
template <typename Idx, typename DType, bool atomic>
constexpr DType Sum<Idx, DType, atomic>::zero;
template <typename Idx, typename DType, bool atomic>
constexpr bool Sum<Idx, DType, atomic>::require_arg;
template <typename Idx,
typename DType,
......@@ -149,6 +174,10 @@ struct Max {
}
}
};
template <typename Idx, typename DType, bool atomic>
constexpr DType Max<Idx, DType, atomic>::zero;
template <typename Idx, typename DType, bool atomic>
constexpr bool Max<Idx, DType, atomic>::require_arg;
template <typename Idx,
typename DType,
......@@ -182,6 +211,10 @@ struct Min {
}
}
};
template <typename Idx, typename DType, bool atomic>
constexpr DType Min<Idx, DType, atomic>::zero;
template <typename Idx, typename DType, bool atomic>
constexpr bool Min<Idx, DType, atomic>::require_arg;
} // namespace reduce
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment