lightning-thunder icon indicating copy to clipboard operation
lightning-thunder copied to clipboard

Published 20 hours ago • verified publisher icon Lightning-AI

Enable `take` and `take_along_axis` in nvfuser executor

Open crcrpar opened this issue 6 months ago • 4 comments

What does this PR do?

Let nvfuser executor run take and take_along_axis. Also, remove the skipif decorator as the min nvfuser version is now 0.2.8 > 0.0.3 as per https://github.com/Lightning-AI/lightning-thunder/blob/5c0127fd7bd11fbe25858f72097e1f12af37adef/thunder/executors/nvfuserex.py#L32-L33

rel:

  • https://github.com/NVIDIA/Fuser/issues/4414
  • https://github.com/NVIDIA/Fuser/issues/4415

crcrpar avatar May 03 '25 04:05 crcrpar