Instructions to use replicate/triton_kernels with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- Kernels
How to use replicate/triton_kernels with Kernels:
# !pip install kernels from kernels import get_kernel # a version (or an explicit revision) is required; see the "Files and versions" tab for the available ones kernel = get_kernel("replicate/triton_kernels", version=1) - Notebooks
- Google Colab
- Kaggle
Upload folder using huggingface_hub
Browse files
build/torch-universal/triton_kernels/__init__.py
CHANGED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from .matmul_ogs import FlexCtx, InFlexData, PrecisionConfig
|
| 2 |
+
|
| 3 |
+
__all__ = ["FlexCtx", "InFlexData", "PrecisionConfig"]
|
torch-ext/triton_kernels/__init__.py
CHANGED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
from .matmul_ogs import FlexCtx, InFlexData, PrecisionConfig
|
| 2 |
+
|
| 3 |
+
__all__ = ["FlexCtx", "InFlexData", "PrecisionConfig"]
|