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

Integrate RMM #2609

Merged
merged 5 commits into from
Jan 14, 2021
Merged

Integrate RMM #2609

merged 5 commits into from
Jan 14, 2021

Conversation

mzient
Copy link
Contributor

@mzient mzient commented Jan 12, 2021

Why we need this PR?

Pick one, remove the rest

  • Refactoring to use RAPIDS memory manager

What happened in this PR?

Fill relevant points, put NA otherwise. Replace anything inside []

  • What solution was applied:
    • Use RMM fork with interfaces we need
    • Replace memory_resource and stream_memory_resource with interfaces from RMM
    • Adjust code to acommodate the interfaces
    • Add RMM dependencies to build system
    • Add RMM and ints dependencies to include paths
  • Affected modules and functionalities:
    • build
    • core/memory manager
  • Key points relevant for the review:
    • N/A
  • Validation and testing:
    • N/A
  • Documentation (including examples):
    • N/A

JIRA TASK: [DALI-1782] [DALI-1783]

Signed-off-by: Michał Zientkiewicz <[email protected]>
@mzient mzient requested a review from a team January 12, 2021 18:46
@mzient
Copy link
Contributor Author

mzient commented Jan 12, 2021

!build

@dali-automaton
Copy link
Collaborator

CI MESSAGE: [1975053]: BUILD STARTED

@dali-automaton
Copy link
Collaborator

CI MESSAGE: [1975053]: BUILD FAILED

Use RMM interfaces in dali::mm.

Signed-off-by: Michał Zientkiewicz <[email protected]>
@dali-automaton
Copy link
Collaborator

CI MESSAGE: [1976929]: BUILD STARTED

@dali-automaton
Copy link
Collaborator

CI MESSAGE: [1976929]: BUILD PASSED

##################################################################
# RAPIDS Memory Manager
##################################################################
include_directories(SYSTEM ${PROJECT_SOURCE_DIR}/third_party/rmm/include)
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Can you check if that compiler for QNX, aarch64 and SBSA?

Copy link
Contributor Author

@mzient mzient Jan 13, 2021

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

QNX passed. I see no reason why SBSA would not, but I can try for CUDA 11 sake.

@dali-automaton
Copy link
Collaborator

CI MESSAGE: [1977341]: BUILD STARTED

Comment on lines 47 to 50
ret = ua(bytes + sizeof(sentinel_t), alignment);
detail::write_sentinel<sentinel_t>(ret, bytes, std::forward<Extra>(extra)...);
} else {
ret = upstream_->allocate(extra..., bytes, alignment);
ret = upstream_->allocate(bytes, alignment, std::forward<Extra>(extra)...);
Copy link
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Why use ua in one place and upstream_ directly in the second?

Copy link
Contributor Author

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

It's a bug. It should be ua.

@dali-automaton
Copy link
Collaborator

CI MESSAGE: [1977856]: BUILD STARTED

@dali-automaton
Copy link
Collaborator

CI MESSAGE: [1977935]: BUILD STARTED

Signed-off-by: Michał Zientkiewicz <[email protected]>
@dali-automaton
Copy link
Collaborator

CI MESSAGE: [1977986]: BUILD STARTED

@dali-automaton
Copy link
Collaborator

CI MESSAGE: [1977986]: BUILD FAILED

Signed-off-by: Michał Zientkiewicz <[email protected]>
@dali-automaton
Copy link
Collaborator

CI MESSAGE: [1978196]: BUILD STARTED

@dali-automaton
Copy link
Collaborator

CI MESSAGE: [1978196]: BUILD FAILED

@dali-automaton
Copy link
Collaborator

CI MESSAGE: [1978369]: BUILD STARTED

@dali-automaton
Copy link
Collaborator

CI MESSAGE: [1977935]: BUILD PASSED

@dali-automaton
Copy link
Collaborator

CI MESSAGE: [1978369]: BUILD PASSED

@mzient mzient changed the title [WIP] Integrate RMM Integrate RMM Jan 14, 2021
@mzient mzient merged commit 3847ea7 into NVIDIA:master Jan 14, 2021
@JanuszL JanuszL mentioned this pull request May 19, 2021
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

4 participants