search

minnervva / torchdetscan

This is a tool for finding non-deterministic functions in your pytorch code.
https://github.com/minnervva/torchdetscan
MIT License
1 stars 0 forks source link

EmbeddingBag when mode='max' #29

Closed markcoletti closed 6 months ago

markcoletti commented 6 months ago

torch.nn.EmbeddingBag when attempting to differentiate a CUDA tensor when mode='max'

markcoletti commented 6 months ago 
Linting file: pytorch_embeddingbag.py
Found non-deterministic function EmbeddingBag at line 8, column 16 with mode=max that will be non-deterministic if used with a CUDA tensor