diff --git a/python/triton/ops/__init__.py b/python/triton/ops/__init__.py index ef4974751..425c42d8f 100644 --- a/python/triton/ops/__init__.py +++ b/python/triton/ops/__init__.py @@ -1,5 +1,4 @@ from .conv import _conv, conv from .matmul import _matmul, matmul -from .softmax import _softmax, softmax from .cross_entropy import _cross_entropy, cross_entropy from . import blocksparse \ No newline at end of file