torch-rb 0.12.2 → 0.13.0
Sign up to get free protection for your applications and to get access to all the features.
- checksums.yaml +4 -4
- data/CHANGELOG.md +5 -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/version.rb +1 -1
- metadata +4 -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
|
[![Build Status](https://github.com/ankane/torch.rb/workflows/build/badge.svg?branch=master)](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"
|