We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent aa82461 commit 4031ac6Copy full SHA for 4031ac6
tests/tests_fabric/test_connector.py
@@ -178,6 +178,10 @@ def parse_devices(devices):
178
@staticmethod
179
def get_parallel_devices(devices):
180
return [torch.device("cpu")] * devices
181
+
182
+ @staticmethod
183
+ def get_device() -> str:
184
+ return "cpu"
185
186
187
def auto_device_count() -> int:
tests/tests_pytorch/trainer/connectors/test_accelerator_connector.py
@@ -191,6 +191,10 @@ def parse_devices(devices):
191
192
193
194
195
196
197
198
199
200
0 commit comments