Skip to content

Commit 93bdabe

Browse files
committed
update block_radix_topk.cuh namespace
1 parent 3bfdd75 commit 93bdabe

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

paddle/phi/kernels/funcs/block_radix_topk.cuh

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -22,8 +22,8 @@
2222
#include <cub/util_ptx.cuh>
2323
#include <cub/util_type.cuh>
2424

25-
namespace paddle {
26-
namespace framework {
25+
namespace phi {
26+
namespace funcs {
2727

2828
template <typename KeyT,
2929
int BLOCK_SIZE,
@@ -386,6 +386,6 @@ class BlockRadixTopKRegister {
386386
unsigned int top_k_mask_;
387387
};
388388

389-
}; // namespace framework
390-
}; // namespace paddle
389+
}; // namespace funcs
390+
}; // namespace phi
391391
#endif

paddle/phi/kernels/gpu/weighted_sample_neighbors_kernel.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -125,7 +125,7 @@ __launch_bounds__(BLOCK_SIZE) __global__
125125
bool topk_is_unique;
126126

127127
using BlockRadixSelectT =
128-
paddle::framework::BlockRadixTopKGlobalMemory<float, BLOCK_SIZE, true>;
128+
phi::funcs::BlockRadixTopKGlobalMemory<float, BLOCK_SIZE, true>;
129129
__shared__ typename BlockRadixSelectT::TempStorage share_storage;
130130

131131
BlockRadixSelectT{share_storage}.radixTopKGetThreshold(

0 commit comments

Comments
 (0)