xla
[Fori_loop|While_loop] Placeholder lower torch.while_loop with python dispatch for simple addition test case
#6532
Merged
Overview
Commits
54
Changes
View On GitHub

[Fori_loop|While_loop] Placeholder lower torch.while_loop with python dispatch for simple addition test case #6532

ManfeiBai merged 54 commits into master from while_loop-lowering-with-simplecalculation-dispatch-in-torch
ManfeiBai
ManfeiBai Create fori_loop.py
0a2cc00c
ManfeiBai Update fori_loop.py
c4b43844
ManfeiBai Update init_python_bindings.cpp
8ac0bb5b
ManfeiBai Update init_python_bindings.cpp
0a0f8c66
ManfeiBai Update init_python_bindings.cpp
c62ad7d4
ManfeiBai Update fori_loop.py
f0dd53e6
ManfeiBai Update fori_loop.py
dcf65fc4
ManfeiBai Create test_fori_loop.py
a1ed583d
ManfeiBai Update test_fori_loop.py
cea90628
ManfeiBai Update test_fori_loop.py
7e1ffcbc
ManfeiBai Update test_fori_loop.py
6694c099
ManfeiBai Update test_fori_loop.py
caeeb505
ManfeiBai use code from xb
6a4ad4ff
ManfeiBai only xb test
76a1cfa7
ManfeiBai check original version has used python dispatch or not
694e7976
ManfeiBai check original version has used python dispatch or not again
fc4c8bb1
ManfeiBai add test script for xla
cf6aae9e
ManfeiBai add test script for xla again
04a8eff9
ManfeiBai test with while_loop_dense
9c4ba742
ManfeiBai change dispatchkey
e377b775
ManfeiBai re dispatch
27d66f82
ManfeiBai re dispatch again
82612f12
ManfeiBai check again
0de6baea
ManfeiBai check type
0977c549
ManfeiBai checkpoint to show body/cond hlo
e76484fb
ManfeiBai correct xb
f5dea5d6
ManfeiBai add example code script
a189666d
ManfeiBai add example code script again
12e6886f
ManfeiBai only test tpu
202f1efc
ManfeiBai only test tpu
d6fd9346
ManfeiBai add result value check
8b6df8e3
ManfeiBai add result value check
2db33ab2
ManfeiBai ManfeiBai added fori_loop
ManfeiBai clean code
6d3f3a4d
ManfeiBai try torchxla code
700ed2aa
ManfeiBai try torchxla code again
0f05d761
ManfeiBai try torchxla code again again
d7e54c33
ManfeiBai try torchxla code again again again
1b113087
ManfeiBai add test on CPU/GPU tests
fea71a62
ManfeiBai add test on TPU test trigger's
9a42faf8
ManfeiBai add test in TPU CI workflow
0d21e476
ManfeiBai Merge branch 'master' into while_loop-lowering-with-simplecalculation…
45c4433d
ManfeiBai ManfeiBai changed the title [Fori_loop|While_loop] Lower torch.while_loop with python dispatch in torch for simple addition test case [Fori_loop|While_loop] Lower torch.while_loop with python dispatch for simple addition test case 1 year ago
ManfeiBai placeholder for xlacomputation
5050eb4f
ManfeiBai add test
92b69f44
ManfeiBai modif test code
c4ed61aa
ManfeiBai modify fori_loop
1c8395ec
ManfeiBai format
c9e7b5fd
ManfeiBai format
02f738fd
ManfeiBai format
d689acc3
ManfeiBai format
122d6e5d
ManfeiBai ManfeiBai force pushed from f35899ac to 122d6e5d 1 year ago
ManfeiBai ManfeiBai marked this pull request as ready for review 1 year ago
ManfeiBai ManfeiBai requested a review from will-cromar will-cromar 1 year ago
ManfeiBai ManfeiBai requested a review from mbzomowski mbzomowski 1 year ago
ManfeiBai ManfeiBai requested a review from miladm miladm 1 year ago
ManfeiBai ManfeiBai requested a review from qihqi qihqi 1 year ago
ManfeiBai ManfeiBai requested a review from yeounoh yeounoh 1 year ago
ManfeiBai ManfeiBai requested a review from JackCaoG JackCaoG 1 year ago
ManfeiBai
mbzomowski
mbzomowski commented on 2024-02-20
ManfeiBai test log
e95ddc3a
qihqi
qihqi approved these changes on 2024-02-21
ManfeiBai test
589b1b0c
ManfeiBai trigger test again
02caa424
yeounoh
yeounoh commented on 2024-02-22
yeounoh
yeounoh commented on 2024-02-22
yeounoh
yeounoh commented on 2024-02-22
ManfeiBai comment to explain
ab5ae68f
yeounoh
yeounoh approved these changes on 2024-02-22
ManfeiBai comment for explain
594280c1
ManfeiBai ManfeiBai changed the title [Fori_loop|While_loop] Lower torch.while_loop with python dispatch for simple addition test case [Fori_loop|While_loop] Placeholder lower torch.while_loop with python dispatch for simple addition test case 1 year ago
ManfeiBai ManfeiBai merged 5405c4d4 into master 1 year ago
ManfeiBai ManfeiBai added q1_2024_manfei

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
fori_loop
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum