PR #7173 Integrate dlpack to dynamo.

Integrate dlpack to dynamo. #7173

vanbasten23 merged 14 commits into master from xiowei/integrate_dlpack_with_dynamo_fallback

vanbasten23 changed the title ~~[DO NOT REVIEW YET] Integrate dlpack to dynamo.~~ Integrate dlpack to dynamo. 1 year ago

vanbasten23 requested a review from

JackCaoG 1 year ago

vanbasten23 requested a review from

ysiraichi 1 year ago

vanbasten23 marked this pull request as ready for review 1 year ago

ysiraichi commented on 2024-06-10

vanbasten23 requested a review from

ysiraichi 1 year ago

Add the test for training.

38b4be47

Integrate the dlpack with dynamo.

4e83fa58

add more stuff

1e4f5ba5

the test passed

1e599c97

remove mark_step in _move_xla_cuda_tensor_to_cuda. It's only needed t…

bd0cb856

add back the assert in pjrt_computation_client

30c087d7

fix a test

f0432198

fix up

46b24414

refine comments

0f3218c8

vanbasten23 force pushed from 5b475102 to 0f3218c8 1 year ago

JackCaoG commented on 2024-06-11

when the from_dlpack's input cuda tensor has required_grad=True, thn…

6b13135f

need mark_step when converting xla tensor to cuda tensor so that the …

2dd2bf19

vanbasten23 requested a review from

JackCaoG 1 year ago

Use cuda back on more tests.

41284a50

make sure there is no fallback.

8ba5d4b6

ysiraichi approved these changes on 2024-06-13

fix comments

18b4dfa1

vanbasten23 merged c216d26c into master 1 year ago

miladm added xla:gpu

miladm assigned

vanbasten23 308 days ago

Reviewers

ysiraichi

JackCaoG

Assignees

vanbasten23

Labels

xla:gpu

Milestone

No milestone

xla Integrate dlpack to dynamo. #7173 Merged

Integrate dlpack to dynamo. #7173

xla
Integrate dlpack to dynamo.
#7173

Merged