[MoE] Use int input for histc on CUDA to support deterministic algorithms #43583
[MoE] Use int input for histc on CUDA/XPU to support deterministic al…
2495c3bb
Update src/transformers/integrations/moe.py
524fc3dc
Merge branch 'main' into moe-deterministic
874c97f3
vasqu
merged
704f7d94
into main 127 days ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub