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

Numba function operator, docs update #2972

Merged
merged 5 commits into from
May 25, 2021

Conversation

majra20
Copy link
Contributor

@majra20 majra20 commented May 18, 2021

Signed-off-by: Rafal Maj rmaj@nvidia.com

Why we need this PR?

Pick one, remove the rest

  • It fixes a bug bug description
  • It adds new feature needed because of why we need this feature
  • Refactoring to improve what

What happened in this PR?

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

  • What solution was applied:
    [ Explain solution of the problem, a new feature added here. ]
  • Affected modules and functionalities:
    [ Describe here what was changed, added, removed. ]
  • Key points relevant for the review:
    [ Describe here what is the most important part that reviewers should focus on. ]
  • Validation and testing:
    [ Describe here if and how this PR is tested. ]
  • Documentation (including examples):
    [ Describe here if documentation and examples were updated. ]

JIRA TASK: [Use DALI-XXXX or NA]

@@ -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
Copy link
Contributor

@JanuszL JanuszL May 18, 2021

Choose a reason for hiding this comment

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

Suggested change
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

@JanuszL JanuszL self-assigned this May 18, 2021
@majra20 majra20 force-pushed the numba_func_batch_processing_doc branch from bb47c12 to 5af3a1e Compare May 19, 2021 07:12
@majra20
Copy link
Contributor Author

majra20 commented May 19, 2021

!build

@dali-automaton
Copy link
Collaborator

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);
Copy link
Contributor

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?

Copy link
Contributor

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.

Copy link
Contributor Author

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.

@dali-automaton
Copy link
Collaborator

CI MESSAGE: [2385958]: BUILD PASSED

@JanuszL JanuszL changed the base branch from master to main May 19, 2021 12:22
Rafal Maj added 4 commits May 20, 2021 09:45
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>
@majra20 majra20 force-pushed the numba_func_batch_processing_doc branch from 521a3be to 9a68f04 Compare May 20, 2021 08:12
@majra20
Copy link
Contributor Author

majra20 commented May 20, 2021

!build

@dali-automaton
Copy link
Collaborator

CI MESSAGE: [2390222]: BUILD STARTED

@dali-automaton
Copy link
Collaborator

CI MESSAGE: [2390222]: BUILD PASSED

Signed-off-by: Rafal Maj <rmaj@nvidia.com>
@majra20
Copy link
Contributor Author

majra20 commented May 25, 2021

!build

@dali-automaton
Copy link
Collaborator

CI MESSAGE: [2405397]: BUILD STARTED

@dali-automaton
Copy link
Collaborator

CI MESSAGE: [2405397]: BUILD PASSED

@majra20 majra20 merged commit d55255e into NVIDIA:main May 25, 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.

None yet

5 participants