kernel
danieldk HF Staff commited on
Commit
d595dec
·
1 Parent(s): 0a590ce
Files changed (41) hide show
  1. build/torch25-cxx11-cu118-x86_64-linux/rotary/_ops.py +3 -3
  2. build/torch25-cxx11-cu118-x86_64-linux/rotary/{_rotary_5yzc45v7kk3yu.abi3.so → _rotary_c7fa2df.abi3.so} +2 -2
  3. build/torch25-cxx11-cu121-x86_64-linux/rotary/_ops.py +3 -3
  4. build/torch25-cxx11-cu121-x86_64-linux/rotary/{_rotary_tbiepw2a2ep3e.abi3.so → _rotary_c7fa2df.abi3.so} +2 -2
  5. build/torch25-cxx11-cu124-x86_64-linux/rotary/_ops.py +3 -3
  6. build/torch25-cxx11-cu124-x86_64-linux/rotary/{_rotary_6w5syhrhmerj6.abi3.so → _rotary_c7fa2df.abi3.so} +2 -2
  7. build/torch25-cxx98-cu118-x86_64-linux/rotary/_ops.py +3 -3
  8. build/torch25-cxx98-cu118-x86_64-linux/rotary/{_rotary_joujmbgvsytzg.abi3.so → _rotary_c7fa2df.abi3.so} +2 -2
  9. build/torch25-cxx98-cu121-x86_64-linux/rotary/_ops.py +3 -3
  10. build/torch25-cxx98-cu121-x86_64-linux/rotary/_rotary_c7fa2df.abi3.so +3 -0
  11. build/torch25-cxx98-cu121-x86_64-linux/rotary/_rotary_mi2o7e7sishyw.abi3.so +0 -3
  12. build/torch25-cxx98-cu124-x86_64-linux/rotary/_ops.py +3 -3
  13. build/torch25-cxx98-cu124-x86_64-linux/rotary/_rotary_c7fa2df.abi3.so +3 -0
  14. build/torch25-cxx98-cu124-x86_64-linux/rotary/_rotary_rngiohfhfwuge.abi3.so +0 -3
  15. build/torch26-cxx11-cu118-x86_64-linux/rotary/_ops.py +3 -3
  16. build/torch26-cxx11-cu118-x86_64-linux/rotary/_rotary_alv7mzltcxxpq.abi3.so +0 -3
  17. build/torch26-cxx11-cu118-x86_64-linux/rotary/_rotary_c7fa2df.abi3.so +3 -0
  18. build/torch26-cxx11-cu124-x86_64-linux/rotary/_ops.py +3 -3
  19. build/torch26-cxx11-cu124-x86_64-linux/rotary/_rotary_c4eyapeep6gty.abi3.so +0 -3
  20. build/torch26-cxx11-cu124-x86_64-linux/rotary/_rotary_c7fa2df.abi3.so +3 -0
  21. build/torch26-cxx11-cu126-x86_64-linux/rotary/_ops.py +3 -3
  22. build/torch26-cxx11-cu126-x86_64-linux/rotary/_rotary_c7fa2df.abi3.so +3 -0
  23. build/torch26-cxx11-cu126-x86_64-linux/rotary/_rotary_lodp6xeztste6.abi3.so +0 -3
  24. build/torch26-cxx98-cu118-x86_64-linux/rotary/_ops.py +3 -3
  25. build/torch26-cxx98-cu118-x86_64-linux/rotary/_rotary_c7fa2df.abi3.so +3 -0
  26. build/torch26-cxx98-cu118-x86_64-linux/rotary/_rotary_z27mls7mz4e7m.abi3.so +0 -3
  27. build/torch26-cxx98-cu124-x86_64-linux/rotary/_ops.py +3 -3
  28. build/torch26-cxx98-cu124-x86_64-linux/rotary/_rotary_3bktke4p3hz3a.abi3.so +0 -3
  29. build/torch26-cxx98-cu124-x86_64-linux/rotary/_rotary_c7fa2df.abi3.so +3 -0
  30. build/torch26-cxx98-cu126-x86_64-linux/rotary/_ops.py +3 -3
  31. build/torch26-cxx98-cu126-x86_64-linux/rotary/_rotary_c7fa2df.abi3.so +3 -0
  32. build/torch26-cxx98-cu126-x86_64-linux/rotary/_rotary_fvednlzeqgg5s.abi3.so +0 -3
  33. build/torch27-cxx11-cu118-x86_64-linux/rotary/__init__.py +19 -0
  34. build/torch27-cxx11-cu118-x86_64-linux/rotary/_ops.py +9 -0
  35. build/torch27-cxx11-cu118-x86_64-linux/rotary/_rotary_c7fa2df.abi3.so +3 -0
  36. build/torch27-cxx11-cu126-x86_64-linux/rotary/__init__.py +19 -0
  37. build/torch27-cxx11-cu126-x86_64-linux/rotary/_ops.py +9 -0
  38. build/torch27-cxx11-cu126-x86_64-linux/rotary/_rotary_c7fa2df.abi3.so +3 -0
  39. build/torch27-cxx11-cu128-x86_64-linux/rotary/__init__.py +19 -0
  40. build/torch27-cxx11-cu128-x86_64-linux/rotary/_ops.py +9 -0
  41. build/torch27-cxx11-cu128-x86_64-linux/rotary/_rotary_c7fa2df.abi3.so +3 -0
build/torch25-cxx11-cu118-x86_64-linux/rotary/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _rotary_5yzc45v7kk3yu
3
- ops = torch.ops._rotary_5yzc45v7kk3yu
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_rotary_5yzc45v7kk3yu::{op_name}"
 
1
  import torch
2
+ from . import _rotary_c7fa2df
3
+ ops = torch.ops._rotary_c7fa2df
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_rotary_c7fa2df::{op_name}"
build/torch25-cxx11-cu118-x86_64-linux/rotary/{_rotary_5yzc45v7kk3yu.abi3.so → _rotary_c7fa2df.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:10612c2e12298dc582d6b5bb4f5c03f0c22de9ce9dc1c2722e0ec8a46380234c
3
- size 4006264
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:08c0da33fb14e91656e99200ed9805cced2f2f447aabef772a9c589f4f324b76
3
+ size 4010344
build/torch25-cxx11-cu121-x86_64-linux/rotary/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _rotary_tbiepw2a2ep3e
3
- ops = torch.ops._rotary_tbiepw2a2ep3e
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_rotary_tbiepw2a2ep3e::{op_name}"
 
1
  import torch
2
+ from . import _rotary_c7fa2df
3
+ ops = torch.ops._rotary_c7fa2df
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_rotary_c7fa2df::{op_name}"
build/torch25-cxx11-cu121-x86_64-linux/rotary/{_rotary_tbiepw2a2ep3e.abi3.so → _rotary_c7fa2df.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:011fa4e0d1ab3b26f540a6265c1d875ed32e6bfa676ee5a511cb498c03e88f46
3
- size 3971536
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:abc6d5b66947918e5cd788887bf70f3f9361133ed0664136e747efae01bdee2f
3
+ size 3975608
build/torch25-cxx11-cu124-x86_64-linux/rotary/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _rotary_6w5syhrhmerj6
3
- ops = torch.ops._rotary_6w5syhrhmerj6
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_rotary_6w5syhrhmerj6::{op_name}"
 
1
  import torch
2
+ from . import _rotary_c7fa2df
3
+ ops = torch.ops._rotary_c7fa2df
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_rotary_c7fa2df::{op_name}"
build/torch25-cxx11-cu124-x86_64-linux/rotary/{_rotary_6w5syhrhmerj6.abi3.so → _rotary_c7fa2df.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:ac12c6e9353d38c4284d7f336e33a7fb10ce8baa41fd664d19d83c44bea6ef45
3
- size 4037768
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:18d7a3eeafe21f995e3e6809a33c1dfc3dbf7803292436806e9621849de5c7ce
3
+ size 4041848
build/torch25-cxx98-cu118-x86_64-linux/rotary/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _rotary_joujmbgvsytzg
3
- ops = torch.ops._rotary_joujmbgvsytzg
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_rotary_joujmbgvsytzg::{op_name}"
 
1
  import torch
2
+ from . import _rotary_c7fa2df
3
+ ops = torch.ops._rotary_c7fa2df
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_rotary_c7fa2df::{op_name}"
build/torch25-cxx98-cu118-x86_64-linux/rotary/{_rotary_joujmbgvsytzg.abi3.so → _rotary_c7fa2df.abi3.so} RENAMED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3678ca0157d21b376dcba6e797c2668bfb6de23aae45f5596cbe37b1e4758977
3
- size 3994592
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:7fdbeadd675610724cf7372d2305772e3c22e78c376b22e682101da85e0aec9b
3
+ size 4002768
build/torch25-cxx98-cu121-x86_64-linux/rotary/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _rotary_mi2o7e7sishyw
3
- ops = torch.ops._rotary_mi2o7e7sishyw
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_rotary_mi2o7e7sishyw::{op_name}"
 
1
  import torch
2
+ from . import _rotary_c7fa2df
3
+ ops = torch.ops._rotary_c7fa2df
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_rotary_c7fa2df::{op_name}"
build/torch25-cxx98-cu121-x86_64-linux/rotary/_rotary_c7fa2df.abi3.so ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9efd8d083931561c7f0c812f5b5b81582dc0bcec959d31c4b2298b280c04408a
3
+ size 3968080
build/torch25-cxx98-cu121-x86_64-linux/rotary/_rotary_mi2o7e7sishyw.abi3.so DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:58ad7b1d6576bf63104b4bc0d621e4db61f7f0cca90f53300a76210f85e75647
3
- size 3964000
 
 
 
 
build/torch25-cxx98-cu124-x86_64-linux/rotary/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _rotary_rngiohfhfwuge
3
- ops = torch.ops._rotary_rngiohfhfwuge
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_rotary_rngiohfhfwuge::{op_name}"
 
1
  import torch
2
+ from . import _rotary_c7fa2df
3
+ ops = torch.ops._rotary_c7fa2df
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_rotary_c7fa2df::{op_name}"
build/torch25-cxx98-cu124-x86_64-linux/rotary/_rotary_c7fa2df.abi3.so ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:021e7ce0398a2faebd1f1f6fbb72131fad7fd439189236d7e44570f34d473ce0
3
+ size 4030248
build/torch25-cxx98-cu124-x86_64-linux/rotary/_rotary_rngiohfhfwuge.abi3.so DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:0e356b8577926664f666a7f2116301350f187796f8eff0ee2455ab557f8fef82
3
- size 4030264
 
 
 
 
build/torch26-cxx11-cu118-x86_64-linux/rotary/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _rotary_alv7mzltcxxpq
3
- ops = torch.ops._rotary_alv7mzltcxxpq
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_rotary_alv7mzltcxxpq::{op_name}"
 
1
  import torch
2
+ from . import _rotary_c7fa2df
3
+ ops = torch.ops._rotary_c7fa2df
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_rotary_c7fa2df::{op_name}"
build/torch26-cxx11-cu118-x86_64-linux/rotary/_rotary_alv7mzltcxxpq.abi3.so DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:ed401bda2448c0b5f7ba58134502044ff84174c821cb043bc0347abb88b5c032
3
- size 4002064
 
 
 
 
build/torch26-cxx11-cu118-x86_64-linux/rotary/_rotary_c7fa2df.abi3.so ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a76f944386307a45545e90f3f951468750a0d391cd95413a5b760f86559b76c4
3
+ size 4006144
build/torch26-cxx11-cu124-x86_64-linux/rotary/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _rotary_c4eyapeep6gty
3
- ops = torch.ops._rotary_c4eyapeep6gty
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_rotary_c4eyapeep6gty::{op_name}"
 
1
  import torch
2
+ from . import _rotary_c7fa2df
3
+ ops = torch.ops._rotary_c7fa2df
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_rotary_c7fa2df::{op_name}"
build/torch26-cxx11-cu124-x86_64-linux/rotary/_rotary_c4eyapeep6gty.abi3.so DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:e1b95eacddea0ae6a837ded34f0e6aa5493e1b89e21440a591e4947d84e5a49b
3
- size 4029480
 
 
 
 
build/torch26-cxx11-cu124-x86_64-linux/rotary/_rotary_c7fa2df.abi3.so ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:f9143d74f09805c03756fdff24a2ac4544a67616fbdbbe0935eda780a8e40437
3
+ size 4033552
build/torch26-cxx11-cu126-x86_64-linux/rotary/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _rotary_lodp6xeztste6
3
- ops = torch.ops._rotary_lodp6xeztste6
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_rotary_lodp6xeztste6::{op_name}"
 
1
  import torch
2
+ from . import _rotary_c7fa2df
3
+ ops = torch.ops._rotary_c7fa2df
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_rotary_c7fa2df::{op_name}"
build/torch26-cxx11-cu126-x86_64-linux/rotary/_rotary_c7fa2df.abi3.so ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4589a109c4961abe7328d3276b629e656888a6153840c70a3a01a2bdffed762a
3
+ size 4043464
build/torch26-cxx11-cu126-x86_64-linux/rotary/_rotary_lodp6xeztste6.abi3.so DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:9d63ff54a10967f2ddde757581332486ca3be1de0b5f36005427ac9845e99d6b
3
- size 4039392
 
 
 
 
build/torch26-cxx98-cu118-x86_64-linux/rotary/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _rotary_z27mls7mz4e7m
3
- ops = torch.ops._rotary_z27mls7mz4e7m
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_rotary_z27mls7mz4e7m::{op_name}"
 
1
  import torch
2
+ from . import _rotary_c7fa2df
3
+ ops = torch.ops._rotary_c7fa2df
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_rotary_c7fa2df::{op_name}"
build/torch26-cxx98-cu118-x86_64-linux/rotary/_rotary_c7fa2df.abi3.so ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2beaeb22730d93d21cfb163ce88ba0162ea79a709e156bbaa1ecad72f50b3c7b
3
+ size 3990432
build/torch26-cxx98-cu118-x86_64-linux/rotary/_rotary_z27mls7mz4e7m.abi3.so DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:78e48f0e972b845d7667b1e1e6c807d8ccbe0e12442d227240a066ab430951a5
3
- size 3990448
 
 
 
 
build/torch26-cxx98-cu124-x86_64-linux/rotary/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _rotary_3bktke4p3hz3a
3
- ops = torch.ops._rotary_3bktke4p3hz3a
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_rotary_3bktke4p3hz3a::{op_name}"
 
1
  import torch
2
+ from . import _rotary_c7fa2df
3
+ ops = torch.ops._rotary_c7fa2df
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_rotary_c7fa2df::{op_name}"
build/torch26-cxx98-cu124-x86_64-linux/rotary/_rotary_3bktke4p3hz3a.abi3.so DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:fb5710d06ef1c68ecde248253c92eea5073ca98411b270accdd13f73470bdc66
3
- size 4022072
 
 
 
 
build/torch26-cxx98-cu124-x86_64-linux/rotary/_rotary_c7fa2df.abi3.so ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:36b4ffa7751d03ee8d75d30dd3c3bfac4f0133e9a0c7beca16acfb4213500080
3
+ size 4026152
build/torch26-cxx98-cu126-x86_64-linux/rotary/_ops.py CHANGED
@@ -1,9 +1,9 @@
1
  import torch
2
- from . import _rotary_fvednlzeqgg5s
3
- ops = torch.ops._rotary_fvednlzeqgg5s
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
- return f"_rotary_fvednlzeqgg5s::{op_name}"
 
1
  import torch
2
+ from . import _rotary_c7fa2df
3
+ ops = torch.ops._rotary_c7fa2df
4
 
5
  def add_op_namespace_prefix(op_name: str):
6
  """
7
  Prefix op by namespace.
8
  """
9
+ return f"_rotary_c7fa2df::{op_name}"
build/torch26-cxx98-cu126-x86_64-linux/rotary/_rotary_c7fa2df.abi3.so ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:4beb22ae8bdc549ee6b560a7c0eb8df5be1efd7047b95c951c6b7c5a877cec23
3
+ size 4036064
build/torch26-cxx98-cu126-x86_64-linux/rotary/_rotary_fvednlzeqgg5s.abi3.so DELETED
@@ -1,3 +0,0 @@
1
- version https://git-lfs.github.com/spec/v1
2
- oid sha256:923d3a0436f6279d7709a48cf30b98598243706cbd14fe1010361c2d2a1791de
3
- size 4031984
 
 
 
 
build/torch27-cxx11-cu118-x86_64-linux/rotary/__init__.py ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from typing import Tuple
2
+ import torch
3
+
4
+ from ._ops import ops
5
+
6
+
7
+ def apply_rotary(
8
+ x1: torch.Tensor,
9
+ x2: torch.Tensor,
10
+ cos: torch.Tensor,
11
+ sin: torch.Tensor,
12
+ out1: torch.Tensor,
13
+ out2: torch.Tensor,
14
+ conj: bool,
15
+ ):
16
+ ops.apply_rotary(x1, x2, cos, sin, out1, out2, conj)
17
+
18
+
19
+ __all__ = ["apply_rotary"]
build/torch27-cxx11-cu118-x86_64-linux/rotary/_ops.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ from . import _rotary_c7fa2df
3
+ ops = torch.ops._rotary_c7fa2df
4
+
5
+ def add_op_namespace_prefix(op_name: str):
6
+ """
7
+ Prefix op by namespace.
8
+ """
9
+ return f"_rotary_c7fa2df::{op_name}"
build/torch27-cxx11-cu118-x86_64-linux/rotary/_rotary_c7fa2df.abi3.so ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:711ef3d4191a529f288bd5763e7159e7ee582b249609476458c350b551fad1cf
3
+ size 5628256
build/torch27-cxx11-cu126-x86_64-linux/rotary/__init__.py ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from typing import Tuple
2
+ import torch
3
+
4
+ from ._ops import ops
5
+
6
+
7
+ def apply_rotary(
8
+ x1: torch.Tensor,
9
+ x2: torch.Tensor,
10
+ cos: torch.Tensor,
11
+ sin: torch.Tensor,
12
+ out1: torch.Tensor,
13
+ out2: torch.Tensor,
14
+ conj: bool,
15
+ ):
16
+ ops.apply_rotary(x1, x2, cos, sin, out1, out2, conj)
17
+
18
+
19
+ __all__ = ["apply_rotary"]
build/torch27-cxx11-cu126-x86_64-linux/rotary/_ops.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ from . import _rotary_c7fa2df
3
+ ops = torch.ops._rotary_c7fa2df
4
+
5
+ def add_op_namespace_prefix(op_name: str):
6
+ """
7
+ Prefix op by namespace.
8
+ """
9
+ return f"_rotary_c7fa2df::{op_name}"
build/torch27-cxx11-cu126-x86_64-linux/rotary/_rotary_c7fa2df.abi3.so ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a37b130c80679a3cfdc2a327f603efff9a2602660375df362ba9e31ba78fa2e5
3
+ size 5620456
build/torch27-cxx11-cu128-x86_64-linux/rotary/__init__.py ADDED
@@ -0,0 +1,19 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from typing import Tuple
2
+ import torch
3
+
4
+ from ._ops import ops
5
+
6
+
7
+ def apply_rotary(
8
+ x1: torch.Tensor,
9
+ x2: torch.Tensor,
10
+ cos: torch.Tensor,
11
+ sin: torch.Tensor,
12
+ out1: torch.Tensor,
13
+ out2: torch.Tensor,
14
+ conj: bool,
15
+ ):
16
+ ops.apply_rotary(x1, x2, cos, sin, out1, out2, conj)
17
+
18
+
19
+ __all__ = ["apply_rotary"]
build/torch27-cxx11-cu128-x86_64-linux/rotary/_ops.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ import torch
2
+ from . import _rotary_c7fa2df
3
+ ops = torch.ops._rotary_c7fa2df
4
+
5
+ def add_op_namespace_prefix(op_name: str):
6
+ """
7
+ Prefix op by namespace.
8
+ """
9
+ return f"_rotary_c7fa2df::{op_name}"
build/torch27-cxx11-cu128-x86_64-linux/rotary/_rotary_c7fa2df.abi3.so ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1d5b19ca007a834051dc1ec8024b10915cdaea2bd74a8f9174c75b85a9a535c9
3
+ size 7514880