Update infra_triggers.tf
6dccf0a0
Skeleton trition support
98281236
Merge branch 'master' into triton
99bf48da
Fix bugs
b89e5588
Fix custom call invocation
64189bd2
miladm
commented
on 2024-03-21
Refactor to include gpu custom call and create triton dir
0c208ef7
Lint fixes
b553ba73
python lint fix
c5129e67
miladm
commented
on 2024-03-22
Updated base image for CI
48e71272
Update github workflow gcr image
e04fc971
Merge branch 'master' into custom
37bf1275
Remove xrt build and test file
6061895d
Add temporary test to run triton kernel
f59ddbf6
Fix tests
158aed44
Update payload for xla gpu custom call
87b92c5f
Update gpu runner
847ccc50
Merge branch 'master' into triton
eca6d523
Extract payload from triton kernel programatically
2348ca35
Merge branch 'master' into triton
110c8c6b
Lint fixes
a2261508
Only build triton files for GPU
4c1f4f5d
build pytorch for ampere gpus
431f822e
c++ lint fix
4bade16c
Python lint fix
1c5b47d9
Fix torch cuda arch list
3138a92b
Use a bigger machine for CI build
3f00cfd5
Add triton test to run_tests.sh
e729cfb7
Update triton env variable
8e304c0f
Set up a separate CI for triton tests
27bdc3a7
Fix github workflow to add _triton.yml
9a3ef849
Rebuild torch xla for triton tests
ade444dd
Create a separate CI tab for triton tests
cb0bb85d
Separate build and test phase for triton
015b1ad9
Fix flags for docker run container
a18028ab
Update triton.yml to output docker image
993ee92c
vanbasten23
commented
on 2024-04-23
Add a python binding to register custom calls and remove jax files
a87b782b
Fix lint
bf05d1b8
Merge main
4582fe8f
Merge master
9680167a
Merge master after updating
a7b94c6d
Update CI to use cuda plugin
e14636ad
will-cromar
commented
on 2024-05-10
Install jaxlib while setting up triton tests
256d8194
Install triton package while running triton tests
c616e644
Experimental: Build pytorch with cuda
60b8d18e
Revert build pytorch with CUDA
2bde624d
Merge branch 'master' into triton
e6c4e0aa
Remove ansible path for triton CI
14ee5455
alanwaketan
commented
on 2024-05-16
Style fixes
25acb268
[Experimental] test new CI
6b0ac187
[Experimental] Set XLA_CUDA=0 for cuda arch in ansible
4d971504
[Experimental] Update CI to build pytorch cuda with ansible
e0790492
Update CI
d9c89b66
Fix CI workflow file
7a6c8091
Fix CI workflow
6b1954d3
Fix the wheels installed for tests requiring torch cuda
21797a62
Add compute_capability=8.6 for xla cuda plugin
e6e89d3e
update TORCH_CUDA_ARCH_LIST
ac45fe18
Experimental build torch and torch_xla cuda wheels
f828fbba
Merge branch 'master' into triton
ac56c005
Update build_and_test.yml
c3b86532
Update dlpack test to only use one device
a1168c60
Remove compute capability 8.6 from cuda plugin
39551a25
Remove triton.sh
35e08695
Default empty torch_cuda_arch_list in ansible config
f95d8982
Merge branch 'master' into triton
291104d2
will-cromar
commented
on 2024-06-06
vanbasten23
commented
on 2024-06-06
Revert CI changes
f5c9b1a4
Revert CI changes pt2
5b23969a
Reviewers
Assignees
Labels
Triton
Milestone
No milestone
Login to write a write a comment.
Login via GitHub