Skip to content

win cuda126 #74990

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged
merged 1 commit into from
Aug 30, 2025
Merged

win cuda126 #74990

Changes from all commits
Commits
Show all changes
1 commit
Select commit
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
15 changes: 2 additions & 13 deletions paddle/phi/kernels/funcs/shuffle_batch.cu.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -149,18 +149,8 @@ struct write_output_op_fixed {
std::uint64_t m;
InputIterT in;
OutputIterT out;
// flag contains inclusive scan of valid keys
// perform gather using valid keys
#if CUDA_VERSION >= 12060 && CUDA_VERSION < 12090 && defined(_WIN32)
_CCCL_EXEC_CHECK_DISABLE
_CCCL_HOST_DEVICE std::size_t operator()(key_flag_tuple_fixed x) {
if (x.key < m) {
// -1 because inclusive scan
out[x.flag - 1] = in[x.key];
}
return 0; // Discarded
}
#else
// flag contains inclusive scan of valid keys
// perform gather using valid keys
__thrust_exec_check_disable__ __host__ __device__ std::size_t operator()(
key_flag_tuple_fixed x) {
if (x.key < m) {
Expand All @@ -169,7 +159,6 @@ struct write_output_op_fixed {
}
return 0; // Discarded
}
#endif
};

template <typename ExecutionPolicy,
Expand Down