diff --git a/master/_modules/index.html b/master/_modules/index.html index e47b0e8ee74..083df7abad6 100644 --- a/master/_modules/index.html +++ b/master/_modules/index.html @@ -265,7 +265,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/_modules/torch_xla/core/xla_model.html b/master/_modules/torch_xla/core/xla_model.html index 202ac9f2cb1..ac1f7c14425 100644 --- a/master/_modules/torch_xla/core/xla_model.html +++ b/master/_modules/torch_xla/core/xla_model.html @@ -265,7 +265,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
@@ -1722,9 +1722,10 @@

Source code for torch_xla.core.xla_model

         if sharding and tensor.dim() > 0 and (tensor.size()[0] %
                                               local_runtime_device_count) != 0:
           raise RuntimeError(
-              "When minibatch is configured, the per-host batch size must be divisible "
-              + "by local runtime device count. Per host input data shape " +
-              f"= {tensor.size()}, local_runtime_device_count = {local_runtime_device_count}"
+              "When minibatch is configured, batch dimension of the tensor " +
+              "must be divisible by local runtime device count.input data shape "
+              +
+              f"={tensor.size()}, local_runtime_device_count = {local_runtime_device_count}"
           )
 
     xtensors = torch_xla._XLAC._xla_tensors_from_aten(tensors, devices,
diff --git a/master/_modules/torch_xla/debug/metrics.html b/master/_modules/torch_xla/debug/metrics.html
index 545563fff23..f785c7076dd 100644
--- a/master/_modules/torch_xla/debug/metrics.html
+++ b/master/_modules/torch_xla/debug/metrics.html
@@ -265,7 +265,7 @@
               
               
                 
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/_modules/torch_xla/distributed/parallel_loader.html b/master/_modules/torch_xla/distributed/parallel_loader.html index b4ad4d16b62..c6042d2880c 100644 --- a/master/_modules/torch_xla/distributed/parallel_loader.html +++ b/master/_modules/torch_xla/distributed/parallel_loader.html @@ -265,7 +265,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/_modules/torch_xla/distributed/spmd/xla_sharding.html b/master/_modules/torch_xla/distributed/spmd/xla_sharding.html index 2a178726554..b739f845d86 100644 --- a/master/_modules/torch_xla/distributed/spmd/xla_sharding.html +++ b/master/_modules/torch_xla/distributed/spmd/xla_sharding.html @@ -265,7 +265,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/_modules/torch_xla/distributed/xla_multiprocessing.html b/master/_modules/torch_xla/distributed/xla_multiprocessing.html index d361a261c97..b76bcec4244 100644 --- a/master/_modules/torch_xla/distributed/xla_multiprocessing.html +++ b/master/_modules/torch_xla/distributed/xla_multiprocessing.html @@ -265,7 +265,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/_modules/torch_xla/experimental/eager.html b/master/_modules/torch_xla/experimental/eager.html index 5904737234a..e7040717034 100644 --- a/master/_modules/torch_xla/experimental/eager.html +++ b/master/_modules/torch_xla/experimental/eager.html @@ -265,7 +265,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/_modules/torch_xla/runtime.html b/master/_modules/torch_xla/runtime.html index 06594d9629d..79b7872959f 100644 --- a/master/_modules/torch_xla/runtime.html +++ b/master/_modules/torch_xla/runtime.html @@ -265,7 +265,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/_modules/torch_xla/torch_xla.html b/master/_modules/torch_xla/torch_xla.html index dd6ef337f64..fd604226448 100644 --- a/master/_modules/torch_xla/torch_xla.html +++ b/master/_modules/torch_xla/torch_xla.html @@ -265,7 +265,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/accelerators/gpu.html b/master/accelerators/gpu.html index f5e1180b9c0..8a35d707a09 100644 --- a/master/accelerators/gpu.html +++ b/master/accelerators/gpu.html @@ -267,7 +267,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/accelerators/tpu.html b/master/accelerators/tpu.html index 54a6292a3d8..f10556658c0 100644 --- a/master/accelerators/tpu.html +++ b/master/accelerators/tpu.html @@ -267,7 +267,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/contribute/bazel.html b/master/contribute/bazel.html index bdadfbb9155..569bc132781 100644 --- a/master/contribute/bazel.html +++ b/master/contribute/bazel.html @@ -266,7 +266,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/contribute/codegen_migration.html b/master/contribute/codegen_migration.html index 107d05eb811..a2cb1f8c561 100644 --- a/master/contribute/codegen_migration.html +++ b/master/contribute/codegen_migration.html @@ -267,7 +267,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/contribute/configure-environment.html b/master/contribute/configure-environment.html index f898c715642..76e09fcdc10 100644 --- a/master/contribute/configure-environment.html +++ b/master/contribute/configure-environment.html @@ -267,7 +267,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/contribute/op_lowering.html b/master/contribute/op_lowering.html index 57893f0c5c7..c6478678bf5 100644 --- a/master/contribute/op_lowering.html +++ b/master/contribute/op_lowering.html @@ -267,7 +267,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/contribute/plugins.html b/master/contribute/plugins.html index d51f036440b..4cc7ae8e6c6 100644 --- a/master/contribute/plugins.html +++ b/master/contribute/plugins.html @@ -267,7 +267,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/features/distop.html b/master/features/distop.html index b1b9f9129b7..925bc623327 100644 --- a/master/features/distop.html +++ b/master/features/distop.html @@ -265,7 +265,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/features/pallas.html b/master/features/pallas.html index 6b0c1a6fdf3..284ec92784b 100644 --- a/master/features/pallas.html +++ b/master/features/pallas.html @@ -267,7 +267,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/features/stablehlo.html b/master/features/stablehlo.html index b554fbdc1c8..b6cfb0fa8a8 100644 --- a/master/features/stablehlo.html +++ b/master/features/stablehlo.html @@ -267,7 +267,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/features/triton.html b/master/features/triton.html index c65d5ec6f4f..4c632fd41a4 100644 --- a/master/features/triton.html +++ b/master/features/triton.html @@ -267,7 +267,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/genindex.html b/master/genindex.html index 1f7b8a0bc73..f40aa2091bb 100644 --- a/master/genindex.html +++ b/master/genindex.html @@ -265,7 +265,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/index.html b/master/index.html index 3b388ce58cc..85ee8f42439 100644 --- a/master/index.html +++ b/master/index.html @@ -266,7 +266,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/learn/api-guide.html b/master/learn/api-guide.html index e71c934107a..ee2166ba634 100644 --- a/master/learn/api-guide.html +++ b/master/learn/api-guide.html @@ -267,7 +267,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/learn/dynamic_shape.html b/master/learn/dynamic_shape.html index f83c510cd98..5cfe31f4bab 100644 --- a/master/learn/dynamic_shape.html +++ b/master/learn/dynamic_shape.html @@ -267,7 +267,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/learn/eager.html b/master/learn/eager.html index 86a9aa7b2de..4316a5dbd13 100644 --- a/master/learn/eager.html +++ b/master/learn/eager.html @@ -267,7 +267,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/learn/pjrt.html b/master/learn/pjrt.html index c160c8fbc4a..c521d7a0f3c 100644 --- a/master/learn/pjrt.html +++ b/master/learn/pjrt.html @@ -267,7 +267,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/learn/pytorch-on-xla-devices.html b/master/learn/pytorch-on-xla-devices.html index 1c32f014e2e..36b9b5ab8aa 100644 --- a/master/learn/pytorch-on-xla-devices.html +++ b/master/learn/pytorch-on-xla-devices.html @@ -267,7 +267,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/learn/troubleshoot.html b/master/learn/troubleshoot.html index 31ecbdf3b92..16bd653d01a 100644 --- a/master/learn/troubleshoot.html +++ b/master/learn/troubleshoot.html @@ -267,7 +267,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/learn/xla-overview.html b/master/learn/xla-overview.html index 1eaac5ad551..2c3231270fa 100644 --- a/master/learn/xla-overview.html +++ b/master/learn/xla-overview.html @@ -267,7 +267,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/notes/source_of_recompilation.html b/master/notes/source_of_recompilation.html index 5a0d2e6bd0f..4ecc9ab896c 100644 --- a/master/notes/source_of_recompilation.html +++ b/master/notes/source_of_recompilation.html @@ -265,7 +265,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/objects.inv b/master/objects.inv index 80d7e63189c..b06afb4a2d4 100644 Binary files a/master/objects.inv and b/master/objects.inv differ diff --git a/master/perf/amp.html b/master/perf/amp.html index 7384e5d9352..8440dbfd40a 100644 --- a/master/perf/amp.html +++ b/master/perf/amp.html @@ -267,7 +267,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/perf/ddp.html b/master/perf/ddp.html index 071c0438bf7..baa7874f34d 100644 --- a/master/perf/ddp.html +++ b/master/perf/ddp.html @@ -267,7 +267,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/perf/dynamo.html b/master/perf/dynamo.html index fa76062a19f..803d67b9403 100644 --- a/master/perf/dynamo.html +++ b/master/perf/dynamo.html @@ -267,7 +267,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/perf/fori_loop.html b/master/perf/fori_loop.html index 240446256f2..407e3d3afb0 100644 --- a/master/perf/fori_loop.html +++ b/master/perf/fori_loop.html @@ -267,7 +267,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/perf/fsdp.html b/master/perf/fsdp.html index 5883efedb95..770a5ec3ee3 100644 --- a/master/perf/fsdp.html +++ b/master/perf/fsdp.html @@ -267,7 +267,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/perf/fsdpv2.html b/master/perf/fsdpv2.html index 4dd4804e6ec..6cf28ed9819 100644 --- a/master/perf/fsdpv2.html +++ b/master/perf/fsdpv2.html @@ -267,7 +267,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/perf/quantized_ops.html b/master/perf/quantized_ops.html index 64e7196323a..b4c18a4ba95 100644 --- a/master/perf/quantized_ops.html +++ b/master/perf/quantized_ops.html @@ -267,7 +267,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/perf/recompilation.html b/master/perf/recompilation.html index f7d35ae678a..a19d085f5d9 100644 --- a/master/perf/recompilation.html +++ b/master/perf/recompilation.html @@ -267,7 +267,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/perf/spmd_advanced.html b/master/perf/spmd_advanced.html index d9d480ffdda..29271a5a559 100644 --- a/master/perf/spmd_advanced.html +++ b/master/perf/spmd_advanced.html @@ -267,7 +267,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/perf/spmd_basic.html b/master/perf/spmd_basic.html index b491a50c4cd..7fe05c7aa6f 100644 --- a/master/perf/spmd_basic.html +++ b/master/perf/spmd_basic.html @@ -267,7 +267,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/perf/spmd_distributed_checkpoint.html b/master/perf/spmd_distributed_checkpoint.html index 7e194973203..1902babac95 100644 --- a/master/perf/spmd_distributed_checkpoint.html +++ b/master/perf/spmd_distributed_checkpoint.html @@ -267,7 +267,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/perf/spmd_gpu.html b/master/perf/spmd_gpu.html index 19aeaa66028..39692ff3993 100644 --- a/master/perf/spmd_gpu.html +++ b/master/perf/spmd_gpu.html @@ -267,7 +267,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/py-modindex.html b/master/py-modindex.html index a7c27ad159d..f9c9cbfc1de 100644 --- a/master/py-modindex.html +++ b/master/py-modindex.html @@ -268,7 +268,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )
diff --git a/master/search.html b/master/search.html index f913b3a179a..30b62e180ea 100644 --- a/master/search.html +++ b/master/search.html @@ -265,7 +265,7 @@
- master (2.6.0+git3efe1eb ) + master (2.6.0+gitdc20b2d )