kernel-fun 0.2.0.dev1__py3-none-any.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- kernel_fun/__init__.py +67 -0
- kernel_fun/_common/__init__.py +11 -0
- kernel_fun/_common/cache.py +99 -0
- kernel_fun/_common/compat.py +168 -0
- kernel_fun/_common/support.py +267 -0
- kernel_fun/cconv/__init__.py +24 -0
- kernel_fun/cconv/_kernels/__init__.py +6 -0
- kernel_fun/cconv/_kernels/strip.py +450 -0
- kernel_fun/cconv/_provenance.py +11 -0
- kernel_fun/cconv/ops.py +256 -0
- kernel_fun/kda/__init__.py +23 -0
- kernel_fun/kda/_kernels/__init__.py +7 -0
- kernel_fun/kda/_kernels/bwd_dhu.py +994 -0
- kernel_fun/kda/_kernels/bwd_intra.py +1089 -0
- kernel_fun/kda/_kernels/bwd_intra_triton.py +328 -0
- kernel_fun/kda/_kernels/bwd_scan.py +1105 -0
- kernel_fun/kda/_kernels/bwd_wy.py +320 -0
- kernel_fun/kda/_kernels/bwd_wy_t.py +309 -0
- kernel_fun/kda/_kernels/fwd_intra_triton.py +101 -0
- kernel_fun/kda/_kernels/fwd_state.py +1065 -0
- kernel_fun/kda/_provenance.py +18 -0
- kernel_fun/kda/autograd.py +114 -0
- kernel_fun/kda/chain.py +150 -0
- kernel_fun/kda/ops.py +342 -0
- kernel_fun-0.2.0.dev1.dist-info/METADATA +347 -0
- kernel_fun-0.2.0.dev1.dist-info/RECORD +30 -0
- kernel_fun-0.2.0.dev1.dist-info/WHEEL +4 -0
- kernel_fun-0.2.0.dev1.dist-info/licenses/LICENSE +201 -0
- kernel_fun-0.2.0.dev1.dist-info/licenses/NOTICE +60 -0
- kernel_fun-0.2.0.dev1.dist-info/licenses/THIRD_PARTY_NOTICES.md +175 -0
|
@@ -0,0 +1,175 @@
|
|
|
1
|
+
# Third-party notices
|
|
2
|
+
|
|
3
|
+
kernel-fun
|
|
4
|
+
Copyright © 2026 The Allen Institute for Artificial Intelligence
|
|
5
|
+
|
|
6
|
+
kernel-fun is licensed under the Apache License, Version 2.0 (see `LICENSE`).
|
|
7
|
+
This file lists the third-party open source projects it depends on, with each
|
|
8
|
+
project's license and required notices.
|
|
9
|
+
|
|
10
|
+
**Scope.** This package does not vendor or redistribute any of the projects
|
|
11
|
+
below — every one of them is installed separately by pip (see the
|
|
12
|
+
`dependencies` and `[project.optional-dependencies]` tables in
|
|
13
|
+
`pyproject.toml`). The entries here are attribution for the code kernel-fun
|
|
14
|
+
builds on and calls into.
|
|
15
|
+
|
|
16
|
+
The one exception is flash-linear-attention: several modules under
|
|
17
|
+
`src/kernel_fun/kda/_kernels/` are restructured ports of fla kernels and carry
|
|
18
|
+
that lineage into this repository's own source. `NOTICE` names them module by
|
|
19
|
+
module; the MIT license below is the one that governs those portions.
|
|
20
|
+
|
|
21
|
+
None of the dependencies below are Apache-2.0 licensed, so there is no upstream
|
|
22
|
+
`NOTICE` file to reproduce. If a dependency is added that is Apache-2.0, check
|
|
23
|
+
it for a `NOTICE` and reproduce its contents here.
|
|
24
|
+
|
|
25
|
+
## Runtime dependencies
|
|
26
|
+
|
|
27
|
+
| Component | Version tested against | License |
|
|
28
|
+
| --- | --- | --- |
|
|
29
|
+
| [fla-core](https://github.com/fla-org/flash-linear-attention) | 0.5.2 | MIT |
|
|
30
|
+
| [torch](https://pytorch.org) | 2.11.0+cu130 | BSD-3-Clause |
|
|
31
|
+
| [triton](https://github.com/triton-lang/triton) | 3.6.0 | MIT |
|
|
32
|
+
|
|
33
|
+
## Optional dependencies (`cu13` / `cu12` extras)
|
|
34
|
+
|
|
35
|
+
These ride in with the NVIDIA training base image rather than being installed
|
|
36
|
+
by kernel-fun in production; when absent, the runtime probe falls back to fla.
|
|
37
|
+
|
|
38
|
+
| Component | Version tested against | License |
|
|
39
|
+
| --- | --- | --- |
|
|
40
|
+
| [nvidia-cutlass-dsl](https://github.com/NVIDIA/cutlass) | 4.6.0.dev0 | NVIDIA proprietary |
|
|
41
|
+
| [cuda-python](https://nvidia.github.io/cuda-python/) | 13.3.1 | NVIDIA Software License |
|
|
42
|
+
|
|
43
|
+
---
|
|
44
|
+
|
|
45
|
+
## MIT License
|
|
46
|
+
|
|
47
|
+
### flash-linear-attention (fla-core) — https://github.com/fla-org/flash-linear-attention
|
|
48
|
+
|
|
49
|
+
Copyright © 2023-2026 Songlin Yang, Yu Zhang, Zhiyuan Li
|
|
50
|
+
|
|
51
|
+
### Triton — https://github.com/triton-lang/triton
|
|
52
|
+
|
|
53
|
+
Copyright © 2018-2020 Philippe Tillet
|
|
54
|
+
Copyright © 2020-2022 OpenAI
|
|
55
|
+
|
|
56
|
+
### License text
|
|
57
|
+
|
|
58
|
+
Permission is hereby granted, free of charge, to any person obtaining a copy of
|
|
59
|
+
this software and associated documentation files (the "Software"), to deal in
|
|
60
|
+
the Software without restriction, including without limitation the rights to
|
|
61
|
+
use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of
|
|
62
|
+
the Software, and to permit persons to whom the Software is furnished to do so,
|
|
63
|
+
subject to the following conditions:
|
|
64
|
+
|
|
65
|
+
The above copyright notice and this permission notice shall be included in all
|
|
66
|
+
copies or substantial portions of the Software.
|
|
67
|
+
|
|
68
|
+
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
69
|
+
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, FITNESS
|
|
70
|
+
FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR
|
|
71
|
+
COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER
|
|
72
|
+
IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN
|
|
73
|
+
CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
|
|
74
|
+
|
|
75
|
+
---
|
|
76
|
+
|
|
77
|
+
## BSD 3-Clause "New" or "Revised" License
|
|
78
|
+
|
|
79
|
+
### PyTorch (torch) — https://github.com/pytorch/pytorch
|
|
80
|
+
|
|
81
|
+
From PyTorch:
|
|
82
|
+
|
|
83
|
+
Copyright (c) 2016- Facebook, Inc (Adam Paszke)
|
|
84
|
+
Copyright (c) 2014- Facebook, Inc (Soumith Chintala)
|
|
85
|
+
Copyright (c) 2011-2014 Idiap Research Institute (Ronan Collobert)
|
|
86
|
+
Copyright (c) 2012-2014 Deepmind Technologies (Koray Kavukcuoglu)
|
|
87
|
+
Copyright (c) 2011-2012 NEC Laboratories America (Koray Kavukcuoglu)
|
|
88
|
+
Copyright (c) 2011-2013 NYU (Clement Farabet)
|
|
89
|
+
Copyright (c) 2006-2010 NEC Laboratories America (Ronan Collobert, Leon Bottou, Iain Melvin, Jason Weston)
|
|
90
|
+
Copyright (c) 2006 Idiap Research Institute (Samy Bengio)
|
|
91
|
+
Copyright (c) 2001-2004 Idiap Research Institute (Ronan Collobert, Samy Bengio, Johnny Mariethoz)
|
|
92
|
+
|
|
93
|
+
From Caffe2:
|
|
94
|
+
|
|
95
|
+
Copyright (c) 2016-present, Facebook Inc. All rights reserved.
|
|
96
|
+
|
|
97
|
+
All contributions by Facebook: Copyright (c) 2016 Facebook Inc.
|
|
98
|
+
All contributions by Google: Copyright (c) 2015 Google Inc.
|
|
99
|
+
All contributions by Yangqing Jia: Copyright (c) 2015 Yangqing Jia
|
|
100
|
+
All contributions by Kakao Brain: Copyright 2019-2020 Kakao Brain
|
|
101
|
+
All contributions by Cruise LLC: Copyright (c) 2022 Cruise LLC
|
|
102
|
+
All contributions by Tri Dao: Copyright (c) 2024 Tri Dao
|
|
103
|
+
All contributions by Arm: Copyright (c) 2021, 2023-2025 Arm Limited and/or its affiliates
|
|
104
|
+
All contributions from Caffe: Copyright (c) 2013, 2014, 2015, the respective contributors
|
|
105
|
+
All other contributions: Copyright (c) 2015, 2016 the respective contributors
|
|
106
|
+
|
|
107
|
+
All rights reserved.
|
|
108
|
+
|
|
109
|
+
The full copyright roster, including the Caffe2 per-contributor model, is in
|
|
110
|
+
PyTorch's own `LICENSE`; the terms are:
|
|
111
|
+
|
|
112
|
+
Redistribution and use in source and binary forms, with or without
|
|
113
|
+
modification, are permitted provided that the following conditions are met:
|
|
114
|
+
|
|
115
|
+
1. Redistributions of source code must retain the above copyright notice, this
|
|
116
|
+
list of conditions and the following disclaimer.
|
|
117
|
+
|
|
118
|
+
2. Redistributions in binary form must reproduce the above copyright notice,
|
|
119
|
+
this list of conditions and the following disclaimer in the documentation
|
|
120
|
+
and/or other materials provided with the distribution.
|
|
121
|
+
|
|
122
|
+
3. Neither the names of Facebook, Deepmind Technologies, NYU, NEC Laboratories
|
|
123
|
+
America and IDIAP Research Institute nor the names of its contributors may
|
|
124
|
+
be used to endorse or promote products derived from this software without
|
|
125
|
+
specific prior written permission.
|
|
126
|
+
|
|
127
|
+
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND
|
|
128
|
+
ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED
|
|
129
|
+
WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
|
|
130
|
+
DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR
|
|
131
|
+
ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
|
|
132
|
+
(INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
|
|
133
|
+
LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
|
|
134
|
+
ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
135
|
+
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
|
|
136
|
+
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
137
|
+
|
|
138
|
+
---
|
|
139
|
+
|
|
140
|
+
## NVIDIA proprietary licenses
|
|
141
|
+
|
|
142
|
+
The two components below are **not** open source and are **not** redistributed
|
|
143
|
+
by kernel-fun. They are declared as optional extras (`cu13` / `cu12`) and are
|
|
144
|
+
normally already present in the NVIDIA CUDA training base image, whose EULA
|
|
145
|
+
governs their use. Each is subject to its own NVIDIA license agreement, which
|
|
146
|
+
accompanies the distribution you install.
|
|
147
|
+
|
|
148
|
+
### nvidia-cutlass-dsl — https://github.com/NVIDIA/cutlass
|
|
149
|
+
|
|
150
|
+
Version 4.6.0.dev0. Distributed by NVIDIA under a proprietary license
|
|
151
|
+
("Other/Proprietary License" in the wheel metadata). Note that the CUTLASS
|
|
152
|
+
*source repository* is BSD-3-Clause, but the CuTe DSL wheels — the artifact
|
|
153
|
+
kernel-fun's CuTe kernels are compiled by — are shipped under NVIDIA's own
|
|
154
|
+
terms. Consult the license bundled with the installed package.
|
|
155
|
+
|
|
156
|
+
### cuda-python — https://nvidia.github.io/cuda-python/
|
|
157
|
+
|
|
158
|
+
Version 13.3.1. Distributed by NVIDIA under the NVIDIA Software License
|
|
159
|
+
(`LicenseRef-NVIDIA-SOFTWARE-LICENSE`). Consult the license bundled with the
|
|
160
|
+
installed package.
|
|
161
|
+
|
|
162
|
+
---
|
|
163
|
+
|
|
164
|
+
## Keeping this file current
|
|
165
|
+
|
|
166
|
+
Regenerate the dependency inventory when `pyproject.toml`'s dependency tables
|
|
167
|
+
change, e.g. with `pip-licenses` in the training image:
|
|
168
|
+
|
|
169
|
+
```bash
|
|
170
|
+
pip-licenses --format=markdown --with-urls --with-license-file \
|
|
171
|
+
--packages torch triton fla-core nvidia-cutlass-dsl cuda-python
|
|
172
|
+
```
|
|
173
|
+
|
|
174
|
+
No trained model weights or datasets ship with this package, so there is no
|
|
175
|
+
weights/data attribution section.
|