codegen/generate_functions.rb in torch-rb-0.13.2 vs codegen/generate_functions.rb in torch-rb-0.14.0
- old
+ new
@@ -46,10 +46,15 @@
f.base_name == "nested_tensor" ||
f.base_name == "split_copy" ||
f.base_name == "split_with_sizes_copy" ||
f.base_name == "unbind_copy" ||
# TODO fix LibTorch 1.13 changes
- f.base_name == "native_channel_shuffle"
+ f.base_name == "native_channel_shuffle" ||
+ # TODO fix LibTorch 2.1 changes
+ f.base_name == "sym_size" ||
+ f.base_name == "sym_numel" ||
+ f.base_name == "sym_storage_offset" ||
+ f.base_name == "sym_stride"
end
end
def group_functions(functions)
nn_functions, other_functions = functions.partition { |f| f.python_module == "nn" }