Skip to content

Remove runtime parameter from implicit ensemble tests#358

Merged
karlhigley merged 3 commits into
NVIDIA-Merlin:mainfrom
oliverholworthy:test-implicit-split-test
Jun 1, 2023
Merged

Remove runtime parameter from implicit ensemble tests#358
karlhigley merged 3 commits into
NVIDIA-Merlin:mainfrom
oliverholworthy:test-implicit-split-test

Conversation

@oliverholworthy
Copy link
Copy Markdown
Contributor

Remove runtime parameter from implicit ensemble tests.

Goal. Fix the failing test that runs in our GPU environment on the Implicit op.

These tests currently fail the second time they run. (None, is equivalent to TritonExecutorRuntime, since that's the default). I think this is because there is some memory being managed somewhere (possibly in implicit) that is not being cleaned up completely after the tests complete. Causing a CUDA out-of-memory error.

@oliverholworthy oliverholworthy added this to the Merlin 23.06 milestone May 31, 2023
@oliverholworthy oliverholworthy self-assigned this May 31, 2023
@github-actions
Copy link
Copy Markdown

Documentation preview

https://nvidia-merlin.github.io/systems/review/pr-358

@karlhigley karlhigley merged commit 28ab734 into NVIDIA-Merlin:main Jun 1, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants