提出 #553644: pytorch pytorch (in torch.cuda.nccl.reduce) 2.6.0 Denial of Service情報

タイトルpytorch pytorch (in torch.cuda.nccl.reduce) 2.6.0 Denial of Service
説明When using `torch.cuda.nccl.reduce` with invalid operation codes, the program crashes with `Aborted (core dumped)` instead of raising a `RuntimeError` or validating the input. This bug can be exploited for a denial-of-service attack. # To Reproduce ``` import torch import torch.cuda as cuda from torch.cuda.nccl import reduce def test_bug(): # Checking for unsupported operations unsupported_ops = [0xFF, 0xAA] # Example of invalid NCCL operation codes for op in unsupported_ops: input_tensor = torch.tensor([1.0, 2.0, 3.0], device=f'cuda:0') output_tensor = torch.zeros_like(input_tensor) reduce(inputs=[input_tensor], output=output_tensor, root=0, op=op) if __name__ == "__main__": test_bug() ``` # Output ``` Aborted (core dumped) ```
ソース⚠️ https://github.com/pytorch/pytorch/issues/150836
ユーザー
 Default436352 (UID 81891)
送信2025年04月08日 17:11 (12 月 ago)
モデレーション2025年05月05日 13:53 (27 days later)
ステータス承諾済み
VulDBエントリ307394 [PyTorch 2.6.0+cu124 torch/cuda/nccl.py torch.cuda.nccl.reduce サービス拒否]
ポイント20

Do you need the next level of professionalism?

Upgrade your account now!