search

szagoruyko / pyinn

CuPy fused PyTorch neural networks ops
MIT License
274 stars 38 forks source link

complie with pytorch>1.3 #29

Open boohit opened 1 year ago

boohit commented 1 year ago

Hi，I want to use pyinn when pytorch is greater than 1.3. Can you tell me how to change it? When pytorch is greater than 1.3, the function class only supports static. I have changed the Im2Col class on line 198 and the Col2Im class on line 216 in im2col. py according to the tutorial. I am not sure if I have changed it correctly. Can you help me take a look? Thank you very much! image image

beansofhell commented 11 months ago

did you managed to fix this ?