torch-ddp 0.1.1 → 0.1.2
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 +10 -2
- data/lib/torch/ddp/version.rb +1 -1
- metadata +1 -1
checksums.yaml
CHANGED
|
@@ -1,7 +1,7 @@
|
|
|
1
1
|
---
|
|
2
2
|
SHA256:
|
|
3
|
-
metadata.gz:
|
|
4
|
-
data.tar.gz:
|
|
3
|
+
metadata.gz: cf1fe012a13aed095f9eb6aa3bb32713a42d42c984a2df271a2f5571b8405686
|
|
4
|
+
data.tar.gz: 9a44d9cd25ccc0d56b294910dd9b5c349a79b5468ff8e67935405b30284ed2ef
|
|
5
5
|
SHA512:
|
|
6
|
-
metadata.gz:
|
|
7
|
-
data.tar.gz:
|
|
6
|
+
metadata.gz: 27f09c0880d00540da310ce7904509e1d12c5d087dc8850cc203819e7248ecbe1a1aab5cec33e8dd0415e04932623eb3de7917153a598260a8791c58984bfbb2
|
|
7
|
+
data.tar.gz: 93ff18257628ff81133b30abe12987159f9c7cfe866a91b5b6c161032483ebb087ad5f1c587268fa541f5b18fa6153578efc4333df6cff86883eb0f2692bac55
|
|
@@ -100,7 +100,9 @@ module Torch
|
|
|
100
100
|
|
|
101
101
|
def patch_device_helpers
|
|
102
102
|
Torch::Device.class_eval do
|
|
103
|
-
define_method(:to_s)
|
|
103
|
+
define_method(:to_s) do
|
|
104
|
+
respond_to?(:_str) ? _str : super()
|
|
105
|
+
end
|
|
104
106
|
end
|
|
105
107
|
|
|
106
108
|
unless Torch.const_defined?(:DeviceString)
|
|
@@ -121,7 +123,13 @@ module Torch
|
|
|
121
123
|
Torch::Device.new(device.to_s)
|
|
122
124
|
end
|
|
123
125
|
end
|
|
124
|
-
|
|
126
|
+
device_str =
|
|
127
|
+
if @device.respond_to?(:_str)
|
|
128
|
+
@device._str
|
|
129
|
+
else
|
|
130
|
+
@device.to_s
|
|
131
|
+
end
|
|
132
|
+
super(device_str)
|
|
125
133
|
end
|
|
126
134
|
|
|
127
135
|
def type
|
data/lib/torch/ddp/version.rb
CHANGED