10000 Commits · tjingrant/xla · GitHub

More Web Proxy on the site http://driver.im/

Commits

Commits on Dec 15, 2022

Commits on Dec 7, 2022

Sharding annotation for empty node (#4292)
yeounoh
authored

Commits on Nov 29, 2022

CI requires cmake>=3.18 to install PyTorch (#4247)
yeounoh
authored

Commits on Nov 24, 2022

[SPMD] fixes bugs with AssignIrValue & ExecuteReplicated (#4233)
yeounoh
authored

Commits on Nov 23, 2022

[SPMD] do add instead of matmul in SPMD unit testing (#4237)
yeounoh
authored

Commits on Nov 15, 2022

[SPMD] fix sharding util bugs & implement unit test cases for 5D tile sharding (#4192)
yeounoh
authored

Commits on Nov 8, 2022

[SPMD] Mesh to support custom device order. (#4162)
yeounoh
authored

Commits on Nov 5, 2022

Enable `padded` option for even tensor sharding (#4156)
yeounoh
authored

Commits on Nov 1, 2022

[SPMD] Test sharding_spec with clear_sharding, deepcopy (#4144)
yeounoh
authored

Commits on Oct 28, 2022

[SPMD] Use on_device_shape in ExecuteReplicated (#4139)
yeounoh
authored

Commits on Oct 17, 2022

[SPMD][PoC] compile & execute with PjRt (#3684)

yeounoh
and
will-cromar
authored

Commits on Sep 14, 2022

[SPMD][PoC] InputHandler for input sharding (#3963)
yeounoh
authored

Commits on Sep 8, 2022

[SPMD][PoC] Tensor sharding util (#3962)
yeounoh
authored

Commits on Jul 28, 2022

Use two existing mirrors for download_llvm_raw (#3791)
yeounoh
authored

Commits on Jul 27, 2022

Enable sccache & remote caching (#3781)
yeounoh
authored

Commits on Jul 13, 2022

Commits on Jul 8, 2022

Install jq to docker and push to ECR repo (#3693)
yeounoh
authored

Commits on Jul 7, 2022

Increase cloudbuild timeout to 24000s (#3691)
yeounoh
authored

Commits on Jun 30, 2022

Overlay secondary llvm-raw remote cache without changing OSS tensorflow & llvm p… (#3664)
yeounoh
authored

Commits on Jun 27, 2022

Disable ECR image update (#3667)
yeounoh
authored

Commits on Jun 4, 2022

More conservative IsSparseGather check to favor dense gather on TPU (#3629)
yeounoh
authored

Commits on May 31, 2022

XLA SPMD PoC impl 57AE ementation with XRT (#3476)
yeounoh
authored

Commits on May 26, 2022

Update sparsity check heuristics used in TorchGather (#3610)
yeounoh
authored

Commits on May 21, 2022

Hotfix cloudbuild bazel update (#3595)
yeounoh
authored

Commits on May 20, 2022

Check if XLA_SANDBOX_BUIDL is unset before applying the bazel patch (#3590)
yeounoh
authored

Commits on May 19, 2022

Commits on May 16, 2022

Revert PR#3450 and use sparse_gather in gather (#3566)
yeounoh
authored

Commits on Apr 29, 2022

Commits on Apr 28, 2022

Rotate the nvidia repo key in the base docker image (#3538)
yeounoh
authored

Commits on Apr 8, 2022

CUDA image stable version tag fix (#3480)
yeounoh
authored

0