未验证 提交 b971ba04 编写于 作者: W wenbin 提交者: GitHub

bug fix (#45853)

上级 83cf6758
......@@ -386,7 +386,6 @@ void invokeLayernormShiftPartition(T *out,
}
}
#if CUDA_ARCH_FP16_SUPPORTED(__CUDA_ARCH__)
template <>
void invokeLayernormShiftPartition(half *out,
const half *input,
......@@ -433,7 +432,6 @@ void invokeLayernormShiftPartition(half *out,
eps);
}
}
#endif
template <typename T>
static void convertAndCopy(const std::vector<float> &host, T *dev) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册