ELS-RD / kernl

Kernl lets you run PyTorch transformer models several times faster on GPU with a single line of code, and is designed to be easily hackable.
http://www.kernl.ai
Apache License 2.0
1.53k stars 95 forks source link

fix: correct stride for act inputs in linear layer #279

Closed gaetansnl closed 1 year ago

gaetansnl commented 1 year ago

fix #227

pommedeterresautee commented 1 year ago

test pass

====================================================================================== 2880 passed, 3 skipped, 81 warnings in 8536.58s (2:22:16) =======================================================================================