Skip to content

Commit

Permalink
[gpu]: STFT: STFT officially support dynamic shapes.
Browse files Browse the repository at this point in the history
  • Loading branch information
pkowalc1 committed Dec 12, 2024
1 parent 8826295 commit d60e95f
Showing 1 changed file with 1 addition and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -216,4 +216,4 @@ REGISTER_DEFAULT_IMPLS(unique_gather, OCL_S, OCL_D);
REGISTER_DEFAULT_IMPLS(scaled_dot_product_attention, OCL_S, OCL_D);
REGISTER_DEFAULT_IMPLS(rope, OCL_S, OCL_D);
REGISTER_DEFAULT_IMPLS(search_sorted, OCL_S, OCL_D);
REGISTER_DEFAULT_IMPLS(STFT, OCL_S);
REGISTER_DEFAULT_IMPLS(STFT, OCL_S, OCL_D);

0 comments on commit d60e95f

Please sign in to comment.