transformers
68abf9ba - Add cuda compatibility check for using `grouped_mm` (#45001)

Commit
26 days ago
Add cuda compatibility check for using `grouped_mm` (#45001) * Add cuda compatibility check for grouped_mm * Explicitly check for torch 2.9 too * make style * Add comment --------- Co-authored-by: Ilyas Moutawwakil <57442720+IlyasMoutawwakil@users.noreply.github.com> Co-authored-by: Anton Vlasjuk <73884904+vasqu@users.noreply.github.com>
Author
Parents
Loading