codegen/generate_functions.rb in torch-rb-0.5.0 vs codegen/generate_functions.rb in torch-rb-0.5.1
- old
+ new
@@ -110,10 +110,10 @@
"#{name[3..-1]}?"
else
name
end
- ruby_name = "_#{ruby_name}" if ["size", "stride", "random!"].include?(ruby_name)
+ ruby_name = "_#{ruby_name}" if ["size", "stride", "random!", "stft"].include?(ruby_name)
# cast for Ruby < 2.7 https://github.com/thisMagpie/fftw/issues/22#issuecomment-49508900
cast = RUBY_VERSION.to_f > 2.7 ? "" : "(VALUE (*)(...)) "
"rb_#{def_method}(m, \"#{ruby_name}\", #{cast}#{type}_#{name}, -1);"