issues
search
jax-ml
/
jax-triton
jax-triton contains integrations between JAX and OpenAI Triton
Apache License 2.0
329
stars
32
forks
source link
issues
Newest
Newest
Most commented
Recently updated
Oldest
Least commented
Least recently updated
Integrate Triton up to [50d803cd](https://github.com/openai/triton/commits/50d803cdb4e68910ed663251100e168ea4d2519d
#292
copybara-service[bot]
opened
2 hours ago
0
Add JAX→Triton fp8 type mapping.
#291
copybara-service[bot]
closed
3 days ago
0
Refactors triton_kernel_call_lowering to support both cuda and rocm.
#290
copybara-service[bot]
closed
4 days ago
0
Now that device is used in backend_init_func, it must always be set
#289
copybara-service[bot]
closed
4 days ago
0
Copybara import of the project:
#288
copybara-service[bot]
closed
4 days ago
0
ROCM updates
#287
rahulbatra85
closed
4 days ago
4
Replace jax.lib.xla_bridge.get_backend() with jax.extend.backend.get_backend()
#286
copybara-service[bot]
closed
2 weeks ago
1
Revert due to CI breakages:
#285
copybara-service[bot]
closed
2 weeks ago
0
Stable set of dependencies?
#284
vvvm23
opened
2 weeks ago
0
Update jax-triton metadata to depend on the Triton 3.0.0 release.
#283
hawkinsp
closed
2 weeks ago
1
Integrate Triton up to [b2de88f8](https://github.com/openai/triton/commits/b2de88f89c1ff8082c92165535e48dece55da392)
#282
copybara-service[bot]
closed
1 week ago
1
ROCM updates
#281
rahulbatra85
closed
2 weeks ago
6
Migrated remaining usages of pl.BlockSpec with the old argument order
#280
copybara-service[bot]
closed
2 months ago
0
Migrated ``pl.BlockSpec`` uses in jax_triton to the new argument order
#279
copybara-service[bot]
closed
2 months ago
0
Integrate Triton up to [@4caedc2](https://github.com/openai/triton/commits/4caedc2a51f5c81f7b63212b9bcb76b430102913)
#278
copybara-service[bot]
closed
4 months ago
0
Removed Pallas forwarding modules from jax_triton
#277
copybara-service[bot]
closed
4 months ago
0
Clarify that jax-triton only works on GPU
#276
skye
opened
5 months ago
0
Add option to dump intermediate artifacts for jax-triton compilation
#275
copybara-service[bot]
closed
4 months ago
1
Integrate Triton up to [8e0c7b42](https://github.com/openai/triton/commits/8e0c7b425ac149c43183de966ffa423fd46e4762)
#274
copybara-service[bot]
closed
4 months ago
0
Is jax-triton replaced by Pallas?
#273
gautierronan
closed
3 months ago
2
fix compatibility issues with gpu_triton and tl_ir method access
#272
dtunai
opened
6 months ago
2
cannot import name 'ShapedArrayRef' due to deprecation in jax._src.state
#271
dtunai
opened
6 months ago
0
WIP - testing out JAX fix
#270
copybara-service[bot]
closed
6 months ago
0
Integrate Triton up to [bfb8e413](https://github.com/openai/triton/commits/bfb8e413b075583228c961bdbb65a98dc54d0868)
#269
copybara-service[bot]
closed
6 months ago
0
[jax_triton] Add parameter allowing user to compile for specific compute capability.
#268
copybara-service[bot]
closed
6 months ago
0
Upgrading to triton 3.x
#267
jbuckman
closed
5 months ago
0
Unable to install Jax-Triton
#266
bsaoptima
opened
6 months ago
1
Moved Pallas GPU lowering registartion code into a separate submodule
#265
copybara-service[bot]
closed
6 months ago
0
Import error encountered in jax_triton
#264
egg5154
opened
6 months ago
7
Only print out diagnostic messages if an environment variable is set
#263
copybara-service[bot]
closed
7 months ago
0
Temporarily disabled debug mode in ir.pass_manager.
#262
copybara-service[bot]
closed
7 months ago
0
Integrate Triton up to [657c49e1](https://github.com/openai/triton/commits/657c49e147ec6093a662707139f4cc92da64a34e)
#261
copybara-service[bot]
closed
7 months ago
1
[triton] Add clustering support and test
#260
copybara-service[bot]
closed
6 months ago
1
compile_ttir_to_ptx_inplace no longer requires an IR context
#259
copybara-service[bot]
closed
7 months ago
0
Fixes for Triton integrate
#258
copybara-service[bot]
closed
7 months ago
0
Changes required for AMD/rocm support with Triton
#257
reachtarunhere
opened
7 months ago
0
[triton] Add clustering information to cuLaunchKernelEx
#256
copybara-service[bot]
opened
8 months ago
1
[triton] Pass cluster_dims to TritonKernel and use cuLaunchKernel if size <= 1
#255
copybara-service[bot]
closed
8 months ago
1
[Triton] Re-enable minBitWidth computation in AccelerateMatmul.cpp logic.
#254
copybara-service[bot]
closed
8 months ago
0
Re-enable minBitWidth computation and revert patch to ElementwiseOpToLLVM
#253
copybara-service[bot]
closed
8 months ago
0
Enabling Hopper optimizations in Triton by default.
#252
copybara-service[bot]
closed
5 months ago
0
Enable MMA V3
#251
copybara-service[bot]
closed
8 months ago
0
Import openai/triton from GitHub.
#250
copybara-service[bot]
closed
8 months ago
0
Pallas now uses MLIR Python builders to lower to Triton IR
#249
copybara-service[bot]
closed
8 months ago
0
Check that a kernel can be launched before adding it to candidate kernels
#248
copybara-service[bot]
opened
8 months ago
1
Part of internal Triton integrate
#247
copybara-service[bot]
closed
8 months ago
1
Unable to Install
#246
karan-dalal
opened
8 months ago
2
Triton internal integration
#245
copybara-service[bot]
closed
8 months ago
1
Problems in Installing at HEAD
#244
LeoXinhaoLee
opened
8 months ago
2
Part of internal Triton integrate
#243
copybara-service[bot]
closed
8 months ago
1
Next