torch-ddp 0.1.0 → 0.1.1
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/lib/torch/ddp/monkey_patch.rb +26 -3
- data/lib/torch/ddp/version.rb +1 -1
- metadata +2 -2
checksums.yaml
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
---
|
|
2
2
|
SHA256:
|
|
3
|
-
metadata.gz:
|
|
4
|
-
data.tar.gz:
|
|
3
|
+
metadata.gz: 22fe592695d427f7b20bc8c2ac008e46759a3edd6d0ddad5019aee709e6f9c98
|
|
4
|
+
data.tar.gz: 6d01523fa38e56dd0de452476c6f67514d971eb43ae2eba78a1fb98f6ba0b4a1
|
|
5
5
|
SHA512:
|
|
6
|
-
metadata.gz:
|
|
7
|
-
data.tar.gz:
|
|
6
|
+
metadata.gz: 151cf148508441bad8db78979200e0b83543040df57d01cd5c21bb36703862b32968b08a815212e806c5956420051df7057b9ed16717ec98c760036c4cae57bf
|
|
7
|
+
data.tar.gz: 5711b842e1a9bb4dce28bcd5026488319c4f8fe6f908d22728ce5f2a6aeeac99afd87cf61371ba231f120a15fb15a5af917fa6a30dcfa23728f0629bfa31be5a
|
|
@@ -108,8 +108,20 @@ module Torch
|
|
|
108
108
|
:DeviceString,
|
|
109
109
|
Class.new(String) do
|
|
110
110
|
def initialize(device)
|
|
111
|
-
@device =
|
|
112
|
-
|
|
111
|
+
@device =
|
|
112
|
+
case device
|
|
113
|
+
when Torch::Device
|
|
114
|
+
device
|
|
115
|
+
when String, Symbol
|
|
116
|
+
Torch::Device.new(device.to_s)
|
|
117
|
+
else
|
|
118
|
+
if device.respond_to?(:_str) && device.respond_to?(:type)
|
|
119
|
+
device
|
|
120
|
+
else
|
|
121
|
+
Torch::Device.new(device.to_s)
|
|
122
|
+
end
|
|
123
|
+
end
|
|
124
|
+
super(@device._str)
|
|
113
125
|
end
|
|
114
126
|
|
|
115
127
|
def type
|
|
@@ -124,7 +136,18 @@ module Torch
|
|
|
124
136
|
end
|
|
125
137
|
|
|
126
138
|
Torch::Tensor.class_eval do
|
|
127
|
-
|
|
139
|
+
alias_method :_torch_ddp_original_device, :device unless method_defined?(:_torch_ddp_original_device)
|
|
140
|
+
|
|
141
|
+
define_method(:device) do
|
|
142
|
+
device_obj =
|
|
143
|
+
if respond_to?(:_device)
|
|
144
|
+
_device
|
|
145
|
+
else
|
|
146
|
+
_torch_ddp_original_device
|
|
147
|
+
end
|
|
148
|
+
|
|
149
|
+
Torch::DeviceString.new(device_obj)
|
|
150
|
+
end
|
|
128
151
|
end
|
|
129
152
|
end
|
|
130
153
|
|
data/lib/torch/ddp/version.rb
CHANGED
metadata
CHANGED
|
@@ -1,14 +1,14 @@
|
|
|
1
1
|
--- !ruby/object:Gem::Specification
|
|
2
2
|
name: torch-ddp
|
|
3
3
|
version: !ruby/object:Gem::Version
|
|
4
|
-
version: 0.1.
|
|
4
|
+
version: 0.1.1
|
|
5
5
|
platform: ruby
|
|
6
6
|
authors:
|
|
7
7
|
- Ivan Razuvaev
|
|
8
8
|
autorequire:
|
|
9
9
|
bindir: bin
|
|
10
10
|
cert_chain: []
|
|
11
|
-
date: 2025-12-
|
|
11
|
+
date: 2025-12-16 00:00:00.000000000 Z
|
|
12
12
|
dependencies:
|
|
13
13
|
- !ruby/object:Gem::Dependency
|
|
14
14
|
name: torch-rb
|