ciflow/torchtitan/186653: Make topk deterministic under deterministic algorithms
🔒
https://github.com
«When torch.use_deterministic_algorithms(True) is enabled, make topk resolve tied values with stable index tie-breaking.
CPU topk now compares by value and then index in deterministic mode. CUDA keeps the existing top-k s...»
Automatische Weiterleitung...
1.5s