Skip to content

Commit 4f714b7

Browse files
authored
win cuda126 (#74990)
1 parent 273fe4a commit 4f714b7

File tree

1 file changed

+2
-13
lines changed

1 file changed

+2
-13
lines changed

paddle/phi/kernels/funcs/shuffle_batch.cu.h

Lines changed: 2 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -149,18 +149,8 @@ struct write_output_op_fixed {
149149
std::uint64_t m;
150150
InputIterT in;
151151
OutputIterT out;
152-
// flag contains inclusive scan of valid keys
153-
// perform gather using valid keys
154-
#if CUDA_VERSION >= 12060 && CUDA_VERSION < 12090 && defined(_WIN32)
155-
_CCCL_EXEC_CHECK_DISABLE
156-
_CCCL_HOST_DEVICE std::size_t operator()(key_flag_tuple_fixed x) {
157-
if (x.key < m) {
158-
// -1 because inclusive scan
159-
out[x.flag - 1] = in[x.key];
160-
}
161-
return 0; // Discarded
162-
}
163-
#else
152+
// flag contains inclusive scan of valid keys
153+
// perform gather using valid keys
164154
__thrust_exec_check_disable__ __host__ __device__ std::size_t operator()(
165155
key_flag_tuple_fixed x) {
166156
if (x.key < m) {
@@ -169,7 +159,6 @@ struct write_output_op_fixed {
169159
}
170160
return 0; // Discarded
171161
}
172-
#endif
173162
};
174163

175164
template <typename ExecutionPolicy,

0 commit comments

Comments
 (0)