But here you might run into the no-specialize case, because the grid argument is a DataType, so no specialization. I.e. do_it_many_times must do a dynamic dispatch on momentum_v1, I think, resulting in allocations.
1 Like