issues
search
jax-ml
/
jax-triton
jax-triton contains integrations between JAX and OpenAI Triton
Apache License 2.0
340
stars
36
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
Temporarily disabled debug mode in ir.pass_manager.
#262
copybara-service[bot]
closed
8 months ago
0
Integrate Triton up to [657c49e1](https://github.com/openai/triton/commits/657c49e147ec6093a662707139f4cc92da64a34e)
#261
copybara-service[bot]
closed
8 months ago
1
[triton] Add clustering support and test
#260
copybara-service[bot]
closed
8 months ago
1
compile_ttir_to_ptx_inplace no longer requires an IR context
#259
copybara-service[bot]
closed
9 months ago
0
Fixes for Triton integrate
#258
copybara-service[bot]
closed
9 months ago
0
Changes required for AMD/rocm support with Triton
#257
reachtarunhere
opened
9 months ago
0
[triton] Add clustering information to cuLaunchKernelEx
#256
copybara-service[bot]
closed
2 weeks ago
1
[triton] Pass cluster_dims to TritonKernel and use cuLaunchKernel if size <= 1
#255
copybara-service[bot]
closed
9 months ago
1
[Triton] Re-enable minBitWidth computation in AccelerateMatmul.cpp logic.
#254
copybara-service[bot]
closed
9 months ago
0
Re-enable minBitWidth computation and revert patch to ElementwiseOpToLLVM
#253
copybara-service[bot]
closed
9 months ago
0
Enabling Hopper optimizations in Triton by default.
#252
copybara-service[bot]
closed
7 months ago
0
Enable MMA V3
#251
copybara-service[bot]
closed
9 months ago
0
Import openai/triton from GitHub.
#250
copybara-service[bot]
closed
10 months ago
0
Pallas now uses MLIR Python builders to lower to Triton IR
#249
copybara-service[bot]
closed
10 months ago
0
Check that a kernel can be launched before adding it to candidate kernels
#248
copybara-service[bot]
opened
10 months ago
1
Part of internal Triton integrate
#247
copybara-service[bot]
closed
10 months ago
1
Unable to Install
#246
karan-dalal
opened
10 months ago
2
Triton internal integration
#245
copybara-service[bot]
closed
10 months ago
1
Problems in Installing at HEAD
#244
LeoXinhaoLee
opened
10 months ago
2
Part of internal Triton integrate
#243
copybara-service[bot]
closed
10 months ago
1
Internal fixes for Triton integration
#242
copybara-service[bot]
closed
10 months ago
0
Integration Fixes on top of Triton import
#241
copybara-service[bot]
closed
10 months ago
0
[jax_triton] Allow user to override the custom call target name.
#240
copybara-service[bot]
closed
1 year ago
0
Expose more Triton compiler tuning options.
#239
copybara-service[bot]
closed
1 year ago
0
triton requirement is out of date
#238
ywrt
opened
1 year ago
1
jax_triton: migrate from deprecated jax.linear_util to jax.extend.linear_util
#237
copybara-service[bot]
closed
1 year ago
0
Updated jax.config import
#236
copybara-service[bot]
closed
1 year ago
0
Updated jax.config import
#235
copybara-service[bot]
closed
1 year ago
0
Bumping Triton version
#234
copybara-service[bot]
closed
1 year ago
0
Bumping Triton version
#233
copybara-service[bot]
closed
7 months ago
0
Testing triton integration 2023-09-14
#232
copybara-service[bot]
closed
1 year ago
1
When cuda is not available, the attributes `triton_kernel_call_lib.TritonKernel` and `triton.JITFunction` are missing. This creates issues when trying to import the code, e.g. using colab adhoc impoort.
#231
copybara-service[bot]
closed
1 year ago
1
Testing triton integration 2023-09-14
#230
copybara-service[bot]
closed
1 year ago
1
Upgrade triton dependency to dev20230822 nightly
#229
jon-chuang
closed
2 weeks ago
2
Pallas Operators
#228
srush
opened
1 year ago
0
Align the custom_call implementation in mlir and hlo_helpers.
#227
copybara-service[bot]
closed
1 year ago
0
Fix type annotation for `triton_call` `args`.
#226
copybara-service[bot]
closed
1 year ago
0
XlaRuntimeError when applying FusedAttention
#225
luise1030
opened
1 year ago
0
Running Pallas
#224
mehdiataei
opened
1 year ago
1
More flexible debugging of triton IRs
#223
copybara-service[bot]
opened
1 year ago
1
Internal change.
#222
copybara-service[bot]
closed
2 weeks ago
0
[jax_triton] Always use "triton_kernel_call" as custom call target name.
#221
copybara-service[bot]
closed
1 year ago
0
[jax_triton] Add user-specified `name` field to serialized format.
#220
copybara-service[bot]
closed
1 year ago
0
Pallas upstream is now working?
#219
sh0416
opened
1 year ago
2
Resolve triton-nightly version error
#218
sh0416
closed
1 year ago
2
[JAX] Replace uses of `jnp.array` in types with `jnp.ndarray`.
#217
copybara-service[bot]
closed
1 year ago
0
Update pyproject.toml
#216
sharadmv
closed
1 year ago
0
Hopper Support?
#215
IzzyPutterman
opened
1 year ago
5
Referenced JAX version does not (yet) include jax.experimental.pallas
#214
markschoene
closed
1 year ago
1
[Pallas] Upstream pallas to JAX
#213
copybara-service[bot]
closed
1 year ago
0
Previous
Next