Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Changing tensor creation #220

Merged
merged 8 commits into from
Feb 7, 2025
Merged

Conversation

ajakovljevicTT
Copy link
Contributor

@ajakovljevicTT ajakovljevicTT commented Jan 31, 2025

Ticket

Fixes #202
Fixes #200

Problem description

Since we were given raw pointers to make tensors for in PJRT, we had no control over their ownership. This PR allocates new memory and copies the contents of the pointer, so now PJRT has full control of its memory.

Copy link

github-actions bot commented Jan 31, 2025

TestsPassed ✅Skipped ⚠️Failed
TT-XLA Tests599 ran431 passed168 skipped0 failed
TestResult
No test annotations available

@codecov-commenter
Copy link

codecov-commenter commented Jan 31, 2025

Codecov Report

Attention: Patch coverage is 84.61538% with 4 lines in your changes missing coverage. Please review.

Project coverage is 78.02%. Comparing base (9ee5247) to head (5fcaf3a).
Report is 1 commits behind head on main.

✅ All tests successful. No failed tests found.

Files with missing lines Patch % Lines
src/common/pjrt_implementation/buffer_instance.cc 60.00% 4 Missing ⚠️
Additional details and impacted files
@@            Coverage Diff             @@
##             main     #220      +/-   ##
==========================================
+ Coverage   77.82%   78.02%   +0.20%     
==========================================
  Files          21       21              
  Lines        1028     1042      +14     
==========================================
+ Hits          800      813      +13     
- Misses        228      229       +1     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.

@ajakovljevicTT ajakovljevicTT force-pushed the ajakovljevic/tensor_changing_creation branch 3 times, most recently from 327ef63 to b847be0 Compare February 4, 2025 10:33
inc/common/pjrt_implementation/buffer_instance.h Outdated Show resolved Hide resolved
src/common/pjrt_implementation/buffer_instance.cc Outdated Show resolved Hide resolved
src/common/pjrt_implementation/buffer_instance.cc Outdated Show resolved Hide resolved
src/common/pjrt_implementation/device_instance.cc Outdated Show resolved Hide resolved
inc/common/pjrt_implementation/device_instance.h Outdated Show resolved Hide resolved
inc/common/pjrt_implementation/buffer_instance.h Outdated Show resolved Hide resolved
inc/common/pjrt_implementation/device_instance.h Outdated Show resolved Hide resolved
src/common/pjrt_implementation/device_instance.cc Outdated Show resolved Hide resolved
inc/common/pjrt_implementation/buffer_instance.h Outdated Show resolved Hide resolved
inc/common/pjrt_implementation/buffer_instance.h Outdated Show resolved Hide resolved
inc/common/pjrt_implementation/buffer_instance.h Outdated Show resolved Hide resolved
inc/common/pjrt_implementation/buffer_instance.h Outdated Show resolved Hide resolved
inc/common/pjrt_implementation/buffer_instance.h Outdated Show resolved Hide resolved
inc/common/pjrt_implementation/device_instance.h Outdated Show resolved Hide resolved
src/common/pjrt_implementation/device_instance.cc Outdated Show resolved Hide resolved
inc/common/pjrt_implementation/buffer_instance.h Outdated Show resolved Hide resolved
src/common/pjrt_implementation/buffer_instance.cc Outdated Show resolved Hide resolved
@ajakovljevicTT ajakovljevicTT force-pushed the ajakovljevic/tensor_changing_creation branch 2 times, most recently from 0e27ca6 to 20ff4d5 Compare February 7, 2025 11:51
@ajakovljevicTT ajakovljevicTT force-pushed the ajakovljevic/tensor_changing_creation branch from 20ff4d5 to ed35338 Compare February 7, 2025 11:51
inc/common/pjrt_implementation/buffer_instance.h Outdated Show resolved Hide resolved
inc/common/pjrt_implementation/buffer_instance.h Outdated Show resolved Hide resolved
src/common/pjrt_implementation/buffer_instance.cc Outdated Show resolved Hide resolved
tests/jax/ops/test_convert.py Outdated Show resolved Hide resolved
@ajakovljevicTT ajakovljevicTT merged commit de4e586 into main Feb 7, 2025
7 checks passed
@ajakovljevicTT ajakovljevicTT deleted the ajakovljevic/tensor_changing_creation branch February 7, 2025 14:20
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
6 participants