-
Notifications
You must be signed in to change notification settings - Fork 622
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
Conversation
!build |
CI MESSAGE: [1172983]: BUILD STARTED |
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]; |
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.
Please, make indent aligned properly with the VALUE_SWITCH
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.
done
Signed-off-by: Rafal <Banas.Rafal97@gmail.com>
bf7f80d
to
c528e61
Compare
!build |
CI MESSAGE: [1173023]: BUILD STARTED |
CI MESSAGE: [1173023]: BUILD PASSED |
Why we need this PR?
What happened in this PR?
The operator just runs initializes and runs the kernel
MelFilterBank operator: new file + small changes in existing
SetupImpl and RunImpl
Existing MelFilterBank tests extended to gpu
The operator was already documented
JIRA TASK: DALI-1291