Skip to content
This repository was archived by the owner on Nov 1, 2024. It is now read-only.

Conversation

@wenleix
Copy link
Contributor

@wenleix wenleix commented Aug 27, 2022

Summary:
Pull Request resolved: #480

It originally use memory::getDefaultScopedMemoryPool() since ExecCtx
constructor expects unique_ptr<MemoryPool>.

Since facebookincubator/velox@15eeddb,
ExecCtx now expects MemoryPool*, so we can use
&velox::memory::getProcessDefaultMemoryManager().getRoot() returned by
TorchArrowGlobalStatic::rootMemoryPool.

This makes TorchArrow always use this process default root memory pool.

Reviewed By: dracifer, vancexu

Differential Revision: D39010989

fbshipit-source-id: 3a383b01163e9866500a6fc8b091080b70684c6e

Summary:
Pull Request resolved: #480

It originally use `memory::getDefaultScopedMemoryPool()` since `ExecCtx`
 constructor expects `unique_ptr<MemoryPool>`.

Since facebookincubator/velox@15eeddb,
`ExecCtx` now expects `MemoryPool*`, so we can use
`&velox::memory::getProcessDefaultMemoryManager().getRoot()` returned by
`TorchArrowGlobalStatic::rootMemoryPool`.

This makes TorchArrow always use this process default root memory pool.

Reviewed By: dracifer, vancexu

Differential Revision: D39010989

fbshipit-source-id: 3a383b01163e9866500a6fc8b091080b70684c6e
@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Aug 27, 2022
@wenleix wenleix merged commit 2e06b53 into main Aug 27, 2022
@wenleix wenleix deleted the minor branch August 27, 2022 03:24
Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants