xla
Triton
#6798
Merged
Overview
Commits
67
Changes
View On GitHub
Commits
  • Update infra_triggers.tf
    ManfeiBai committed 2 years ago
  • Skeleton trition support
    bhavya01 committed 1 year ago
  • Merge branch 'master' into triton
    bhavya01 committed 1 year ago
  • Fix bugs
    bhavya01 committed 1 year ago
  • Fix custom call invocation
    bhavya01 committed 1 year ago
  • Refactor to include gpu custom call and create triton dir
    bhavya01 committed 1 year ago
  • Lint fixes
    bhavya01 committed 1 year ago
  • python lint fix
    bhavya01 committed 1 year ago
  • Updated base image for CI
    bhavya01 committed 1 year ago
  • Update github workflow gcr image
    bhavya01 committed 1 year ago
  • Merge branch 'master' into custom
    bhavya01 committed 1 year ago
  • Remove xrt build and test file
    bhavya01 committed 1 year ago
  • Add temporary test to run triton kernel
    bhavya01 committed 1 year ago
  • Fix tests
    bhavya01 committed 1 year ago
  • Update payload for xla gpu custom call
    bhavya01 committed 1 year ago
  • Update gpu runner
    bhavya01 committed 1 year ago
  • Merge branch 'master' into triton
    bhavya01 committed 1 year ago
  • Extract payload from triton kernel programatically
    bhavya01 committed 1 year ago
  • Merge branch 'master' into triton
    bhavya01 committed 1 year ago
  • Lint fixes
    bhavya01 committed 1 year ago
  • Only build triton files for GPU
    bhavya01 committed 1 year ago
  • build pytorch for ampere gpus
    bhavya01 committed 1 year ago
  • c++ lint fix
    bhavya01 committed 1 year ago
  • Python lint fix
    bhavya01 committed 1 year ago
  • Fix torch cuda arch list
    bhavya01 committed 1 year ago
  • Use a bigger machine for CI build
    bhavya01 committed 1 year ago
  • Add triton test to run_tests.sh
    bhavya01 committed 1 year ago
  • Update triton env variable
    bhavya01 committed 1 year ago
  • Set up a separate CI for triton tests
    bhavya01 committed 1 year ago
  • Fix github workflow to add _triton.yml
    bhavya01 committed 1 year ago
  • Rebuild torch xla for triton tests
    bhavya01 committed 1 year ago
  • Create a separate CI tab for triton tests
    bhavya01 committed 1 year ago
  • Separate build and test phase for triton
    bhavya01 committed 1 year ago
  • Fix flags for docker run container
    bhavya01 committed 1 year ago
  • Update triton.yml to output docker image
    bhavya01 committed 1 year ago
  • Add a python binding to register custom calls and remove jax files
    bhavya01 committed 1 year ago
  • Fix lint
    bhavya01 committed 1 year ago
  • Merge main
    bhavya01 committed 1 year ago
  • Merge master
    bhavya01 committed 1 year ago
  • Merge master after updating
    bhavya01 committed 1 year ago
  • Update CI to use cuda plugin
    bhavya01 committed 1 year ago
  • Install jaxlib while setting up triton tests
    bhavya01 committed 1 year ago
  • Install triton package while running triton tests
    bhavya01 committed 1 year ago
  • Experimental: Build pytorch with cuda
    bhavya01 committed 1 year ago
  • Revert build pytorch with CUDA
    bhavya01 committed 1 year ago
  • Merge branch 'master' into triton
    bhavya01 committed 1 year ago
  • Remove ansible path for triton CI
    bhavya01 committed 1 year ago
  • Style fixes
    bhavya01 committed 1 year ago
  • [Experimental] test new CI
    bhavya01 committed 1 year ago
  • [Experimental] Set XLA_CUDA=0 for cuda arch in ansible
    bhavya01 committed 1 year ago
  • [Experimental] Update CI to build pytorch cuda with ansible
    bhavya01 committed 1 year ago
  • Update CI
    bhavya01 committed 1 year ago
  • Fix CI workflow file
    bhavya01 committed 1 year ago
  • Fix CI workflow
    bhavya01 committed 1 year ago
  • Fix the wheels installed for tests requiring torch cuda
    bhavya01 committed 1 year ago
  • Add compute_capability=8.6 for xla cuda plugin
    bhavya01 committed 1 year ago
  • update TORCH_CUDA_ARCH_LIST
    bhavya01 committed 1 year ago
  • Experimental build torch and torch_xla cuda wheels
    bhavya01 committed 1 year ago
  • Merge branch 'master' into triton
    bhavya01 committed 1 year ago
  • Update build_and_test.yml
    bhavya01 committed 1 year ago
  • Update dlpack test to only use one device
    bhavya01 committed 1 year ago
  • Remove compute capability 8.6 from cuda plugin
    bhavya01 committed 1 year ago
  • Remove triton.sh
    bhavya01 committed 1 year ago
  • Default empty torch_cuda_arch_list in ansible config
    bhavya01 committed 1 year ago
  • Merge branch 'master' into triton
    bhavya01 committed 1 year ago
  • Revert CI changes
    bhavya01 committed 1 year ago
  • Revert CI changes pt2
    bhavya01 committed 1 year ago
FAQ Terms Privacy Refunds Impressum
Loading