Skip to content

Commit f88b4eb

Browse files
authored
update block_radix_topk.cuh namespace (#74232)
* fix, test=document_fix * update block_radix_topk.cuh namespace * fix
1 parent a065284 commit f88b4eb

File tree

3 files changed

+4
-25
lines changed

3 files changed

+4
-25
lines changed

paddle/phi/kernels/funcs/block_radix_topk.cuh

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

25-
namespace paddle {
26-
namespace framework {
25+
namespace phi::funcs {
2726

2827
template <typename KeyT,
2928
int BLOCK_SIZE,
@@ -386,6 +385,5 @@ class BlockRadixTopKRegister {
386385
unsigned int top_k_mask_;
387386
};
388387

389-
}; // namespace framework
390-
}; // namespace paddle
388+
} // namespace phi::funcs
391389
#endif

paddle/phi/kernels/gpu/weighted_sample_neighbors_kernel.cu

Lines changed: 2 additions & 2 deletions
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(
@@ -254,7 +254,7 @@ __launch_bounds__(BLOCK_SIZE) __global__
254254
RandomNumGen rng(gidx, random_seed);
255255
float weight_keys[ITEMS_PER_THREAD];
256256
int neighbor_idxs[ITEMS_PER_THREAD];
257-
using BlockRadixTopKT = paddle::framework::
257+
using BlockRadixTopKT = phi::funcs::
258258
BlockRadixTopKRegister<float, BLOCK_SIZE, ITEMS_PER_THREAD, true, int>;
259259
__shared__ typename BlockRadixTopKT::TempStorage sort_tmp_storage;
260260

tools/enforce/count_enforce_by_dir.sh

Lines changed: 0 additions & 19 deletions
Original file line numberDiff line numberDiff line change
@@ -25,30 +25,11 @@
2525

2626
# Result Example:
2727

28-
# paddle/fluid/operators/benchmark | 11 | 0 | 11
2928
# paddle/fluid/operators/collective | 28 | 1 | 27
3029
# paddle/fluid/operators/controlflow | 60 | 59 | 1
31-
# paddle/fluid/operators/detail | 2 | 0 | 2
3230
# paddle/fluid/operators/detection | 276 | 146 | 130
33-
# paddle/fluid/operators/distributed/brpc | 17 | 0 | 17
34-
# paddle/fluid/operators/distributed/grpc | 13 | 0 | 13
35-
# paddle/fluid/operators/distributed | 63 | 10 | 53
36-
# paddle/fluid/operators/distributed_ops | 88 | 6 | 82
37-
# paddle/fluid/operators/elementwise/mkldnn | 5 | 5 | 0
3831
# paddle/fluid/operators/elementwise | 29 | 20 | 9
3932
# paddle/fluid/operators/fused | 227 | 182 | 45
40-
# paddle/fluid/operators/jit/gen | 17 | 0 | 17
41-
# paddle/fluid/operators/jit/more/intrinsic | 0 | 0 | 0
42-
# paddle/fluid/operators/jit/more/mix | 1 | 0 | 1
43-
# paddle/fluid/operators/jit/more/mkl | 9 | 0 | 9
44-
# paddle/fluid/operators/jit/more | 0 | 0 | 0
45-
# paddle/fluid/operators/jit/refer | 8 | 0 | 8
46-
# paddle/fluid/operators/jit | 18 | 0 | 18
47-
# paddle/fluid/operators/lite | 2 | 2 | 0
48-
# paddle/fluid/operators/math/detail | 0 | 0 | 0
49-
# paddle/fluid/operators/math | 200 | 7 | 193
50-
# paddle/fluid/operators/metrics | 38 | 29 | 9
51-
# paddle/fluid/operators/onednn | 107 | 14 | 93
5233
# paddle/fluid/operators/nccl | 27 | 0 | 27
5334
# paddle/fluid/operators/optimizers | 214 | 50 | 164
5435
# paddle/fluid/operators/reader | 40 | 14 | 26

0 commit comments

Comments
 (0)