Code

Fori loop simple case test test#7028

Code

Closed

ManfeiBai wants to merge 53 commits intopytorch:fori_loop_simple_case_test_testpytorch/xla:fori_loop_simple_case_test_testfrom ManfeiBai:fori_loop_simple_case_test_testManfeiBai/PyTorchXLA:fori_loop_simple_case_test_testCopy head branch name to clipboard

+2,234-684

Commits

Commits on Apr 17, 2024

• Add doc explaining how it works (pytorch#6937)

  qihqi

  authored
  f44e1df

• Fix profiling in benchmark script (pytorch#6934)

  vanbasten23

  authored
  310f08d

Commits on Apr 18, 2024

• [Pallas] Integrate FlashAttention with SPMD (pytorch#6935)

  Show description for 9f2b82d
  alanwaketan

  authored
  9f2b82d

Commits on Apr 19, 2024

• Use TPU build for CPU and GPU Python tests (pytorch#6921)

  will-cromar

  authored
  b2556d6

• [Fori_loop|While_loop] Create fori_loop.md (pytorch#6942)

  Show description for 0417d4d
  ManfeiBai

  authored
  0417d4d

• Update XLA pin, 04/19/2024 (pytorch#6944)

  yeounoh

  authored
  2ec7706

• Update jinja and sphinx versions to address the vulnearbility (pytorch#6946)

  yeounoh

  authored
  9ba844a

Commits on Apr 20, 2024

• Make nms fallback by default. (pytorch#6933)

  ysiraichi

  authored
  b06c9c7

Commits on Apr 22, 2024

• revert expand test with dynamo (pytorch#6950)

  bdhirsh

  authored
  62a2b11

• Lower embedding bag forward only (pytorch#6951)

  bhavya01

  authored
  46919a4

Commits on Apr 24, 2024

• Cleanup the code example in the torch_xla2 README. (pytorch#6939)

  dewitt

  authored
  6fd448d

• [torch_xla2] Simplify developer setup steps (pytorch#6905)

  Show description for 89efd17
  

  will-cromar

  and

  qihqi

  authored
  89efd17

• Create rc13 trigger (pytorch#6956)

  lsy323

  authored
  fa090a2

• update torch deps to 2.3 (pytorch#6959)

  qihqi

  authored
  b5574d8

• update rc14 (pytorch#6962)

  lsy323

  authored
  69eeace

• [Doc] Update Pallas user guide (pytorch#6961)

  Show description for a7749fa
  alanwaketan

  authored
  a7749fa

• Add final 2.3 trigger (pytorch#6963)

  lsy323

  authored
  5369e7d

• Temporarily ignore torch commit in CI test (pytorch#6964)

  will-cromar

  authored
  76f7dd0

Commits on Apr 25, 2024

• [Doc] Improve docker instructions (pytorch#6969)

  Show description for af74c34
  alanwaketan

  authored
  af74c34

• Enable PagedAttention through Pallas (pytorch#6912)

  wonjoolee95

  authored
  6ed2026

• Update readme for 2.3 releae (pytorch#6967)

  lsy323

  authored
  0a204a6

• Update GPU readme (pytorch#6968)

  lsy323

  authored
  abe090a

• Write torch_xla.version.__torch_gitrev__ to file directly (pytorch#6966)

  will-cromar

  authored
  0054ec0

• fix pytorch CI after pin update, change test to use assertLessEqual (pytorch#6973)

  bdhirsh

  authored
  2a204e9

• Update Openxla-pin to 04/24 (pytorch#6975)

  ManfeiBai

  authored
  4b48134

Commits on Apr 26, 2024

• Move test_grad_checkpoint.py to tpu test list (pytorch#6976)

  JackCaoG

  authored
  2bf59e0

• Revert "Update Openxla-pin to 04/24" (pytorch#6980)

  ManfeiBai

  authored
  023e2c8

• Update CODEOWNERS for build infrastructure (pytorch#6953)

  will-cromar

  authored
  3f5ff0f

• Move .torch_pin and handle in ansible (pytorch#6920)

  will-cromar

  authored
  b3be775

• Update dynamo test to be less constrain (pytorch#6981)

  JackCaoG

  authored
  b9a9449

• Build CPP tests in new CI workflow (pytorch#6947)

  will-cromar

  authored
  b834e49

• Run TPU CI when label is on PR (pytorch#6984)

  will-cromar

  authored
  174f407

Commits on Apr 29, 2024

• Add readme to call a model (lost due to merge conflicts) (pytorch#6986)

  qihqi

  authored
  6443e59

• Fix permission issues during CI checkout (pytorch#6985)

  will-cromar

  authored
  6d01bb6

• [Revert Revert] Update OpenXLA-pin update to Apr25 (pytorch#6982)

  ManfeiBai

  authored
  971ebe1

• Update test_export_fx_passes.py (pytorch#6972)

  avikchaudhuri

  authored
  7527816

• Change name of CI documentation (pytorch#6994)

  will-cromar

  authored
  7cc78a6

• Rework docs push (pytorch#6954)

  will-cromar

  authored
  c91171d

• sudo rm leftover files in GHA (pytorch#6995)

  will-cromar

  authored
  0e032b1

• Fixes to dynamic_shapes args in test_unbounded_dynamism.py (pytorch#6999)

  avikchaudhuri

  authored
  73b915b

• Manually push to gh-pages branch (pytorch#6996)

  will-cromar

  authored
  d25f475

Commits on Apr 30, 2024

• Re-land: dynamo expand test with view-replay. (pytorch#6958)

  ysiraichi

  authored
  42db709

• Move the nightly whl instruction out of the hide area (pytorch#7000)

  JackCaoG

  authored
  87329ce

• Don't fail docs push if there's nothing to commit (pytorch#7001)

  will-cromar

  authored
  5f75290

• Complain when TensorFlow is installed (pytorch#7004)

  will-cromar

  authored
  4a5e238

• Clean up workspace before test (pytorch#7005)

  will-cromar

  authored
  b8f8fa9

• Tag CI build with git hash (pytorch#7003)

  will-cromar

  authored
  77bbf7f

• fix addbmm opinfo (pytorch#6993)

  qihqi

  authored
  2399e10

Commits on May 1, 2024

• Fix more opinfo tests (pytorch#7008)

  qihqi

  authored
  2907ab3

• Fix q dtype in paged attention kernel (pytorch#7011)

  wonjoolee95

  authored
  865836a

• Build upstream CI image on push to master (pytorch#6952)

  will-cromar

  authored
  4883f6f

• [Pallas] Support segment ids in flash attention (pytorch#6943)

  Show description for 400bd0c
  alanwaketan

  authored
  400bd0c

• Support pin pr number in new .torch_pin (pytorch#6998)

  Show description for cbbefa2
  

  JackCaoG

  and

  will-cromar

  authored
  cbbefa2