Code

Rebase index_copy fix#62

Code

Open

wonjoolee95 wants to merge 4 commits intooptimize_spmd_shardingpytorch-tpu/llama:optimize_spmd_shardingfrom optimize_spmd_sharding_rebase_index_copypytorch-tpu/llama:optimize_spmd_sharding_rebase_index_copyCopy head branch name to clipboard

+2-2

Commits

Commits on May 24, 2024

• Add debug prints

  

  bhavya01

  authored and

  wonjoolee95

  committed
  8df68a9

• Updated indexx_copy to in-place index_copy_

  

  bhavya01

  authored and

  wonjoolee95

  committed
  e07f0bf

• Remove assignment

  

  bhavya01

  authored and

  wonjoolee95

  committed
  1a0a426

• remove extra space

  

  bhavya01

  authored and

  wonjoolee95

  committed
  2b1217e