Code

Add dlpack support#7025

Code

Merged

vanbasten23 merged 19 commits intomasterpytorch/xla:masterfrom xiowei/addDlpackpytorch/xla:xiowei/addDlpackCopy head branch name to clipboardMay 22, 2024

+673-6

Commits

Commits on May 21, 2024

• added the test and the pybinding.

  vanbasten23

  committed
  822d169

• Add toDLPack and it compiles.

  vanbasten23

  committed
  70996c4

• finished fromDLPack and it compiles

  vanbasten23

  committed
  09ff265

• fixed an unrecognized symbol issue.

  vanbasten23

  committed
  1f0ed12

• Change test to use .to(xla_device).

  vanbasten23

  committed
  b3516c2

• check pjrt_buffer is not null. Also in toDlPack, avoid doing from device to host transfer.

  vanbasten23

  committed
  f78d320

• Improved the basic test

  vanbasten23

  committed
  3e2ff67

• add release GIL in to_dlpack.

  vanbasten23

  committed
  01691c0

• fixed a bug of not passing in the pjrt_buffer

  vanbasten23

  committed
  251ae89

• in the middle of fixing holds_[i] == 0 (1 vs. 0)

  vanbasten23

  committed
  aac102b

• fixed the issue: pjrt_stream_executor_client.cc:1322] Check failed: holds_[i] == 0 (1 vs. 0)

  vanbasten23

  committed
  57a936e

• improve the unit tests. All test passed.

  vanbasten23

  committed
  9ff3dd9

• Clean up unused prints and comments

  vanbasten23

  committed
  8dd8381

• linter

  vanbasten23

  committed
  60b4d59

• fix comment

  vanbasten23

  committed
  93dba65

• fix comments

  vanbasten23

  committed
  867dd53

• fix comment

  vanbasten23

  committed
  345945e

• linter

  vanbasten23

  committed
  2e8ff4a

• fix comments and run linter

  vanbasten23

  committed
  98cc739