Code

Support SPMD fsdp compute dtype#13

Code

Merged

lausannel merged 4 commits intoaccAlibabaPAI/xla:accfrom dev/spmd_fsdp_compute_dtypeAlibabaPAI/xla:dev/spmd_fsdp_compute_dtypeCopy head branch name to clipboardSep 20, 2024

+19-4

Commits

Commits on Sep 14, 2024

• feat: support compute dtype in spmd fsdp

  lausannel

  committed
  db59a32

  Copy full SHA for db59a32

Commits on Sep 17, 2024

• fix: import functions from python fsdp

  lausannel

  committed
  f4f9125

  Copy full SHA for f4f9125

Commits on Sep 19, 2024

• fix: unused import

  lausannel

  committed
  4d783f1

  Copy full SHA for 4d783f1

Commits on Sep 20, 2024

• fix: remove comment

  lausannel

  committed
  2741407

  Copy full SHA for 2741407