ARROW-5186 [Plasma] Fix crash caused by improper free on CUDA memory#4177
ARROW-5186 [Plasma] Fix crash caused by improper free on CUDA memory#4177shengjun1985 wants to merge 2 commits intoapache:masterfrom shengjun1985:master
Conversation
Codecov Report
@@ Coverage Diff @@
## master #4177 +/- ##
==========================================
+ Coverage 87.77% 89.18% +1.41%
==========================================
Files 758 617 -141
Lines 92346 82204 -10142
Branches 1251 0 -1251
==========================================
- Hits 81060 73317 -7743
+ Misses 11169 8887 -2282
+ Partials 117 0 -117
Continue to review full report at Codecov.
|
cpp/src/plasma/test/client_tests.cc
Outdated
| AssertCudaRead(object_buffers[0].metadata, {42}); | ||
| } | ||
|
|
||
| TEST_F(TestPlasmaStore, DeleteObjectsGUPTest) { |
cpp/src/plasma/store.cc
Outdated
| #ifdef PLASMA_CUDA | ||
| std::shared_ptr<CudaContext> context_; | ||
| manager_->GetContext(object->device_num - 1, &context_); | ||
| context_->Free(object->pointer, object->data_size + object->metadata_size); |
There was a problem hiding this comment.
Unchecked Status here (this would fail CI if we had GPU-enabled builds). Seems that EraseFromObjectTable will need to return Status now?
|
I'll wait for @pcmoritz to review |
|
@pcmoritz can you review? |
|
Yes, I'm looking at it now! |
pcmoritz
left a comment
There was a problem hiding this comment.
The code looks good to me!
I'll also test it on a CUDA enabled machine, feel free to merge if that has already been done.
|
Tests are passing! Thanks for the contribution :) |
No description provided.