pytorch / extension-cpp

C++ extensions in PyTorch
1.02k stars 214 forks source link

Replace lltm with myaddmul; update to new custom ops APIs #94

Closed zou3519 closed 6 months ago

zou3519 commented 6 months ago

Stack from ghstack (oldest at bottom):

We're replacing lltm with myaddmul(a: Tensor, b: Tensor, c: float), which just does a*b+c. This simplification allows us to focus on the operator registration instead of get lost in the details of the complicated lltm kernels.

Test Plan:

zou3519 commented 6 months ago

Uh that's not how to land it