torch-rb 0.12.1 → 0.13.0
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- checksums.yaml +4 -4
- data/CHANGELOG.md +9 -0
- data/README.md +3 -1
- data/codegen/generate_functions.rb +4 -2
- data/codegen/native_functions.yaml +1392 -593
- data/ext/torch/backends.cpp +4 -0
- data/ext/torch/tensor.cpp +4 -4
- data/ext/torch/utils.h +1 -1
- data/lib/torch/utils/data/data_pipes/filter_iter_data_pipe.rb +37 -0
- data/lib/torch/utils/data/data_pipes/iter/file_lister.rb +74 -0
- data/lib/torch/utils/data/data_pipes/iter/file_opener.rb +51 -0
- data/lib/torch/utils/data/data_pipes/iter/iterable_wrapper.rb +30 -0
- data/lib/torch/utils/data/data_pipes/iter/stream_wrapper.rb +30 -0
- data/lib/torch/utils/data/data_pipes/iter_data_pipe.rb +41 -0
- data/lib/torch/utils/data/iterable_dataset.rb +13 -0
- data/lib/torch/version.rb +1 -1
- data/lib/torch.rb +7 -0
- metadata +11 -4
checksums.yaml
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
---
|
|
2
2
|
SHA256:
|
|
3
|
-
metadata.gz:
|
|
4
|
-
data.tar.gz:
|
|
3
|
+
metadata.gz: 772ef01be10ea497b3dd029c74bd87c650bd2adf2d1f3f26688d9401d2a1d2af
|
|
4
|
+
data.tar.gz: d22bc868344e5d64d70e5327c64dcc3ffe3deacaab4998e157eeec04f27a71de
|
|
5
5
|
SHA512:
|
|
6
|
-
metadata.gz:
|
|
7
|
-
data.tar.gz:
|
|
6
|
+
metadata.gz: b66ec8876a57926a39c2111684205e4110a21af7f9e3dc0c660784f3a27a042fc9395a86bd5cc90bbb99e062116e0e95b7041e2ae9bac2f885b68788a48d97a4
|
|
7
|
+
data.tar.gz: 6417859882a411b2fcb9ab24b694b6934444e4ac38a9f478f4878eac26730de4c220d9bd38e8e728a263422a752ab47c842528462184ea839e157698e4209d51
|
data/CHANGELOG.md
CHANGED
data/README.md
CHANGED
|
@@ -8,6 +8,7 @@ Check out:
|
|
|
8
8
|
- [TorchText](https://github.com/ankane/torchtext) for text and NLP tasks
|
|
9
9
|
- [TorchAudio](https://github.com/ankane/torchaudio) for audio tasks
|
|
10
10
|
- [TorchRec](https://github.com/ankane/torchrec-ruby) for recommendation systems
|
|
11
|
+
- [TorchData](https://github.com/ankane/torchdata-ruby) for data loading
|
|
11
12
|
|
|
12
13
|
[](https://github.com/ankane/torch.rb/actions)
|
|
13
14
|
|
|
@@ -409,7 +410,8 @@ Here’s the list of compatible versions.
|
|
|
409
410
|
|
|
410
411
|
Torch.rb | LibTorch
|
|
411
412
|
--- | ---
|
|
412
|
-
0.
|
|
413
|
+
0.13.0+ | 2.0.0+
|
|
414
|
+
0.12.0-0.12.2 | 1.13.0-1.13.1
|
|
413
415
|
0.11.0-0.11.2 | 1.12.0-1.12.1
|
|
414
416
|
0.10.0-0.10.2 | 1.11.0
|
|
415
417
|
0.9.0-0.9.2 | 1.10.0-1.10.2
|
|
@@ -415,6 +415,8 @@ def generate_function_params(function, params, remove_self)
|
|
|
415
415
|
"memoryformat"
|
|
416
416
|
when "Storage"
|
|
417
417
|
"storage"
|
|
418
|
+
when "Layout"
|
|
419
|
+
"layout"
|
|
418
420
|
else
|
|
419
421
|
raise "Unknown type: #{param[:type]} (#{function.name})"
|
|
420
422
|
end
|
|
@@ -445,7 +447,7 @@ def generate_dispatch_code(function, def_method, params, opt_index, remove_self)
|
|
|
445
447
|
# torch::empty sets requires_grad by at::empty doesn't
|
|
446
448
|
# https://github.com/pytorch/pytorch/issues/36455
|
|
447
449
|
prefix = remove_self ? "self." : (opt_index ? "torch::" : "at::")
|
|
448
|
-
dispatch = function.dispatch_name
|
|
450
|
+
dispatch = nil # function.dispatch_name
|
|
449
451
|
unless dispatch
|
|
450
452
|
dispatch = function.base_name
|
|
451
453
|
dispatch += "_symint" if function.func.include?("SymInt")
|
|
@@ -640,7 +642,7 @@ def signature_type(param)
|
|
|
640
642
|
when "int"
|
|
641
643
|
"int64_t"
|
|
642
644
|
when "SymInt"
|
|
643
|
-
"
|
|
645
|
+
"SymInt"
|
|
644
646
|
when "float"
|
|
645
647
|
"double"
|
|
646
648
|
when "str"
|