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

MelFilterBank GPU operator #1796

Merged
merged 1 commit into from
Mar 6, 2020

Conversation

banasraf
Copy link
Collaborator

@banasraf banasraf commented Mar 6, 2020

Why we need this PR?

  • It adds a gpu MelFilterBank operator

What happened in this PR?

  • What solution was applied:
    The operator just runs initializes and runs the kernel
  • Affected modules and functionalities:
    MelFilterBank operator: new file + small changes in existing
  • Key points relevant for the review:
    SetupImpl and RunImpl
  • Validation and testing:
    Existing MelFilterBank tests extended to gpu
  • Documentation (including examples):
    The operator was already documented

JIRA TASK: DALI-1291

@banasraf
Copy link
Collaborator Author

banasraf commented Mar 6, 2020

!build

@dali-automaton
Copy link
Collaborator

CI MESSAGE: [1172983]: BUILD STARTED

Comment on lines 31 to 37
using MelFilterBankKernel = kernels::audio::MelFilterBankGpu<T, Dims>;
kmgr_.Initialize<MelFilterBankKernel>();
kmgr_.Resize<MelFilterBankKernel>(1, 1);
output_desc[0].type = TypeTable::GetTypeInfo(TypeTable::GetTypeID<T>());
auto in_view = view<const T, Dims>(input);
auto &req = kmgr_.Setup<MelFilterBankKernel>(0, ctx_, in_view, args_);
output_desc[0].shape = req.output_shapes[0];
Copy link
Contributor

Choose a reason for hiding this comment

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

Please, make indent aligned properly with the VALUE_SWITCH

Copy link
Collaborator Author

Choose a reason for hiding this comment

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

done

Signed-off-by: Rafal <Banas.Rafal97@gmail.com>
@banasraf banasraf force-pushed the gpu-mel-filter-bank-operator branch from bf7f80d to c528e61 Compare March 6, 2020 09:15
@banasraf
Copy link
Collaborator Author

banasraf commented Mar 6, 2020

!build

@dali-automaton
Copy link
Collaborator

CI MESSAGE: [1173023]: BUILD STARTED

@dali-automaton
Copy link
Collaborator

CI MESSAGE: [1173023]: BUILD PASSED

@banasraf banasraf merged commit 52655f1 into NVIDIA:master Mar 6, 2020
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