search

csarofeen / pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration
http://pytorch.org
Other
26 stars 7 forks source link

index_select/gather cache patch #2501

Closed jjsjann123 closed 1 year ago

jjsjann123 commented 1 year ago

patching index_select/gather on cache check to examine reduction dimension. (original issue exposed lightning) Added python tests to verify the fix. lintrunner to check FLAKE8 on nvfuser python code

jjsjann123 commented 1 year ago

cc'ing @ftxj

jjsjann123 commented 1 year ago

I'm guessing conflicts are coming from the excessive black clean up. Will resolve when I get a chance.

ftxj commented 1 year ago

cc'ing @ftxj

Know the change, thanks!

jjsjann123 commented 1 year ago

devel is broken. I have the foul PR reverted and is running CI locally. I'll merge once CI finishes.

jjsjann123 commented 1 year ago

CI passed. Merging this one