Skip to content

Commit 96e16c3

Browse files
Snehal-ReddyLegNeato
authored andcommitted
fix(cuda_std): align WarpShuffleMode discriminants with LLVM intrinsics
1 parent 5fec7ae commit 96e16c3

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

crates/cuda_std/src/warp.rs

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -735,10 +735,10 @@ impl WarpShuffleValue for bf16 {
735735
#[repr(u32)]
736736
#[derive(Clone, Copy)]
737737
pub enum WarpShuffleMode {
738-
Up,
739-
Down,
740-
Idx,
741-
Xor,
738+
Idx = 0,
739+
Up = 1,
740+
Down = 2,
741+
Xor = 3,
742742
}
743743

744744
// C-compatible struct to match LLVM IR's {i32, i8} return type

0 commit comments

Comments
 (0)