-
Notifications
You must be signed in to change notification settings - Fork 611
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
Numba function operator, docs update #2972
Conversation
@@ -112,7 +112,8 @@ This function must work in Numba ``nopython`` mode.)code", DALI_PYTHON_OBJECT) | |||
This function is invoked once per batch. Also this function must work in Numba ``nopython`` mode.)code", | |||
DALI_PYTHON_OBJECT, nullptr) | |||
.AddOptionalArg("batch_processing", R"code(Determines whether the function is invoked once per batch or | |||
separately for every sample in the batch.)code", false); | |||
separately for every sample in the batch. For CPU usage we recommend not using this option. When operator |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
separately for every sample in the batch. For CPU usage we recommend not using this option. When operator | |
separately for every sample in the batch. | |
The best performance for the CPU is obtained when this option is turned off. When operator |
bb47c12
to
5af3a1e
Compare
!build |
CI MESSAGE: [2385958]: BUILD STARTED |
@@ -131,7 +134,7 @@ DALI_SCHEMA(NumbaFuncImpl) | |||
.AddOptionalArg<int>("setup_fn", R"code(Address of setup function setting shapes for outputs. | |||
This function is invoked once per batch.)code", 0) | |||
.AddOptionalArg("batch_processing", R"code(Determines whether the function is invoked once per batch or | |||
separately for every sample in the batch.)code", false); | |||
separately for each sample in the batch.)code", false); |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Shouldn't the explanation be here, too?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Perhaps it should be here only.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This isn't shown in our documentation, it has MakeInternal()
option.
CI MESSAGE: [2385958]: BUILD PASSED |
Signed-off-by: Rafal Maj <rmaj@nvidia.com>
Signed-off-by: Rafal Maj <rmaj@nvidia.com>
Signed-off-by: Rafal Maj <rmaj@nvidia.com>
Signed-off-by: Rafal Maj <rmaj@nvidia.com>
521a3be
to
9a68f04
Compare
!build |
CI MESSAGE: [2390222]: BUILD STARTED |
CI MESSAGE: [2390222]: BUILD PASSED |
Signed-off-by: Rafal Maj <rmaj@nvidia.com>
!build |
CI MESSAGE: [2405397]: BUILD STARTED |
CI MESSAGE: [2405397]: BUILD PASSED |
Signed-off-by: Rafal Maj rmaj@nvidia.com
Why we need this PR?
Pick one, remove the rest
What happened in this PR?
Fill relevant points, put NA otherwise. Replace anything inside []
[ Explain solution of the problem, a new feature added here. ]
[ Describe here what was changed, added, removed. ]
[ Describe here what is the most important part that reviewers should focus on. ]
[ Describe here if and how this PR is tested. ]
[ Describe here if documentation and examples were updated. ]
JIRA TASK: [Use DALI-XXXX or NA]