diff --git a/master/_modules/index.html b/master/_modules/index.html index eedc06237b2..2198aac6ecc 100644 --- a/master/_modules/index.html +++ b/master/_modules/index.html @@ -225,7 +225,7 @@
"""Returns a list of supported devices of a given kind.
Args:
- devkind (string..., optional): If specified, one of `TPU`, `GPU`, `XPU`,
+ devkind (string..., optional): If specified, one of `TPU`, `GPU`, `XPU`,
`NEURON` or `CPU` (the 'GPU' XLA device is currently not implemented).
max_devices (int, optional): The maximum number of devices to be returned of
that kind.
@@ -521,7 +521,7 @@ Source code for torch_xla.core.xla_model
n (int, optional): The specific instance (ordinal) to be returned. If
specified, the specific XLA device instance will be returned. Otherwise
the first device of `devkind` will be returned.
- devkind (string..., optional): If specified, one of `TPU`, `CUDA`, `XPU`
+ devkind (string..., optional): If specified, one of `TPU`, `CUDA`, `XPU`
`NEURON`, `ROCM` or `CPU`.
Returns:
@@ -545,7 +545,10 @@ Source code for torch_xla.core.xla_model
return _DEVICES.value[int(m.group(1))]
-def xla_real_devices(devices):
+def xla_real_devices(devices: Optional[List[torch.device]] = None):
+ if not devices:
+ devices = get_xla_supported_devices()
+
return [_xla_real_device(device) for device in devices]
@@ -557,7 +560,7 @@ Source code for torch_xla.core.xla_model
real device.
Returns:
- A string representation of the hardware type (`CPU`, `TPU`, `XPU`, `NEURON`, `GPU`, `CUDA`, `ROCM`)
+ A string representation of the hardware type (`CPU`, `TPU`, `XPU`, `NEURON`, `GPU`, `CUDA`, `ROCM`)
of the given device.
"""
real_device = _xla_real_device(device)
diff --git a/master/_modules/torch_xla/distributed/parallel_loader.html b/master/_modules/torch_xla/distributed/parallel_loader.html
index e9b9e4cdabc..85427250450 100644
--- a/master/_modules/torch_xla/distributed/parallel_loader.html
+++ b/master/_modules/torch_xla/distributed/parallel_loader.html
@@ -225,7 +225,7 @@
- master (2.2.0+git6f3f38f )
+ master (2.2.0+gitad14582 )
diff --git a/master/_modules/torch_xla/distributed/xla_multiprocessing.html b/master/_modules/torch_xla/distributed/xla_multiprocessing.html
index 623600ace95..d1367e05fed 100644
--- a/master/_modules/torch_xla/distributed/xla_multiprocessing.html
+++ b/master/_modules/torch_xla/distributed/xla_multiprocessing.html
@@ -225,7 +225,7 @@
- master (2.2.0+git6f3f38f )
+ master (2.2.0+gitad14582 )
diff --git a/master/_modules/torch_xla/utils/serialization.html b/master/_modules/torch_xla/utils/serialization.html
index 17df1e335ee..a5b897c2a72 100644
--- a/master/_modules/torch_xla/utils/serialization.html
+++ b/master/_modules/torch_xla/utils/serialization.html
@@ -225,7 +225,7 @@
- master (2.2.0+git6f3f38f )
+ master (2.2.0+gitad14582 )
diff --git a/master/_modules/torch_xla/utils/utils.html b/master/_modules/torch_xla/utils/utils.html
index b6db5b28df1..0f2b1e4e88e 100644
--- a/master/_modules/torch_xla/utils/utils.html
+++ b/master/_modules/torch_xla/utils/utils.html
@@ -225,7 +225,7 @@
- master (2.2.0+git6f3f38f )
+ master (2.2.0+gitad14582 )
diff --git a/master/genindex.html b/master/genindex.html
index bb13c33ae31..7d3527f7c16 100644
--- a/master/genindex.html
+++ b/master/genindex.html
@@ -226,7 +226,7 @@
- master (2.2.0+git6f3f38f )
+ master (2.2.0+gitad14582 )
diff --git a/master/index.html b/master/index.html
index f09680cd449..af0d2d4e590 100644
--- a/master/index.html
+++ b/master/index.html
@@ -225,7 +225,7 @@
- master (2.2.0+git6f3f38f )
+ master (2.2.0+gitad14582 )
diff --git a/master/notes/source_of_recompilation.html b/master/notes/source_of_recompilation.html
index bd6b1fdb560..5c71613d4af 100644
--- a/master/notes/source_of_recompilation.html
+++ b/master/notes/source_of_recompilation.html
@@ -225,7 +225,7 @@
- master (2.2.0+git6f3f38f )
+ master (2.2.0+gitad14582 )
diff --git a/master/objects.inv b/master/objects.inv
index b0529682eef..d9156f99f96 100644
Binary files a/master/objects.inv and b/master/objects.inv differ
diff --git a/master/py-modindex.html b/master/py-modindex.html
index f56b130e599..b3922073a38 100644
--- a/master/py-modindex.html
+++ b/master/py-modindex.html
@@ -228,7 +228,7 @@
- master (2.2.0+git6f3f38f )
+ master (2.2.0+gitad14582 )
diff --git a/master/search.html b/master/search.html
index dc3c88e65a2..ccc05675ffa 100644
--- a/master/search.html
+++ b/master/search.html
@@ -225,7 +225,7 @@
- master (2.2.0+git6f3f38f )
+ master (2.2.0+gitad14582 )