Hugging Face's logo

monteirot
/
my_softmax_function

kernel
Model card Files
xet
 Community
tiagomonteiro0715 commited on
Commit
dd87c36
·
1 Parent(s): 32b5efc

sX

Browse files
Files changed (2) hide show
  1. torch-ext/my_softmax_function/__init__.py +1 -1
  2. torch-ext/my_softmax_function/{_ops.py → _kernel.py} +0 -0
torch-ext/my_softmax_function/__init__.py CHANGED
@@ -3,6 +3,6 @@
3
  Public API exposed to `kernels` consumers.
4
  """
5
 
6
- from ._ops import attention, softmax_attention
7
 
8
  __all__ = ["attention", "softmax_attention"]
 
3
  Public API exposed to `kernels` consumers.
4
  """
5
 
6
+ from ._kernel import attention, softmax_attention
7
 
8
  __all__ = ["attention", "softmax_attention"]
torch-ext/my_softmax_function/{_ops.py → _kernel.py} RENAMED
File without changes