未验证 提交 cfaee812 编写于 作者: C chalsliu 提交者: GitHub

Fix jetson compile error (#45692)

上级 a1dbee23
......@@ -68,8 +68,8 @@ static inline int RoundToPowerOfTwo(int n) {
#ifdef WITH_NV_JETSON
// The number of threads cannot be assigned 1024 in some cases when the device
// is nano or tx2 .
template <typename phi::GPUContext>
inline void ChangeThreadNum(const phi::GPUContext& context,
template <typename GPUContext>
inline void ChangeThreadNum(const GPUContext& context,
int* num_thread,
int alternative_num_thread = 512) {
if (context.GetComputeCapability() == 53 ||
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册