未验证 提交 60ba559a 编写于 作者: S sneaxiy 提交者: GitHub

fix win cu116 compile error (#52894)

上级 7aafeb45
......@@ -582,8 +582,10 @@ void ComputeFusedGemmEpilogueBackwardImpl(const phi::GPUContext& dev_ctx,
bool use_addto_dx,
bool use_addto_dy) {
using MT = typename phi::dtype::MPTypeTrait<T>::Type;
static_assert(std::is_same<DXT, T>::value || std::is_same<DXT, MT>::value);
static_assert(std::is_same<DYT, T>::value || std::is_same<DYT, MT>::value);
constexpr bool kIsValidDataType =
(std::is_same<DXT, T>::value || std::is_same<DXT, MT>::value) &&
(std::is_same<DYT, T>::value || std::is_same<DYT, MT>::value);
static_assert(kIsValidDataType, "Invalid data type");
using Trait = FusedGEMMGradTrait<TransX, TransY>;
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册