From 20905c0ff7ba6260235b6bb7c2a873549c39268f Mon Sep 17 00:00:00 2001 From: torchxlabot2 Date: Wed, 10 Jul 2024 00:37:44 +0000 Subject: [PATCH] Update doc from commit 8ac6425742aca73ac7011464fb8924e0efee53f9 --- release/r2.4/_modules/index.html | 2 +- .../_modules/torch_xla/core/xla_model.html | 2 +- .../_modules/torch_xla/debug/metrics.html | 2 +- .../distributed/parallel_loader.html | 2 +- .../distributed/spmd/xla_sharding.html | 2 +- .../distributed/xla_multiprocessing.html | 2 +- .../torch_xla/experimental/eager.html | 19 +++++++++++++++--- release/r2.4/_modules/torch_xla/runtime.html | 2 +- .../r2.4/_modules/torch_xla/torch_xla.html | 2 +- release/r2.4/genindex.html | 2 +- release/r2.4/index.html | 2 +- .../r2.4/notes/source_of_recompilation.html | 2 +- release/r2.4/objects.inv | Bin 871 -> 871 bytes release/r2.4/py-modindex.html | 2 +- release/r2.4/search.html | 2 +- 15 files changed, 29 insertions(+), 16 deletions(-) diff --git a/release/r2.4/_modules/index.html b/release/r2.4/_modules/index.html index 2fb2f586e2a..7a29a8dd09b 100644 --- a/release/r2.4/_modules/index.html +++ b/release/r2.4/_modules/index.html @@ -265,7 +265,7 @@
- master (2.4.0+git18b351f ) + master (2.4.0+git8ac6425 )
diff --git a/release/r2.4/_modules/torch_xla/core/xla_model.html b/release/r2.4/_modules/torch_xla/core/xla_model.html index 84dc7b7f524..2f264d66581 100644 --- a/release/r2.4/_modules/torch_xla/core/xla_model.html +++ b/release/r2.4/_modules/torch_xla/core/xla_model.html @@ -265,7 +265,7 @@
- master (2.4.0+git18b351f ) + master (2.4.0+git8ac6425 )
diff --git a/release/r2.4/_modules/torch_xla/debug/metrics.html b/release/r2.4/_modules/torch_xla/debug/metrics.html index e6ec0b6f7f9..cc103fc38f1 100644 --- a/release/r2.4/_modules/torch_xla/debug/metrics.html +++ b/release/r2.4/_modules/torch_xla/debug/metrics.html @@ -265,7 +265,7 @@
- master (2.4.0+git18b351f ) + master (2.4.0+git8ac6425 )
diff --git a/release/r2.4/_modules/torch_xla/distributed/parallel_loader.html b/release/r2.4/_modules/torch_xla/distributed/parallel_loader.html index 95c4af28530..05dc0f2263c 100644 --- a/release/r2.4/_modules/torch_xla/distributed/parallel_loader.html +++ b/release/r2.4/_modules/torch_xla/distributed/parallel_loader.html @@ -265,7 +265,7 @@
- master (2.4.0+git18b351f ) + master (2.4.0+git8ac6425 )
diff --git a/release/r2.4/_modules/torch_xla/distributed/spmd/xla_sharding.html b/release/r2.4/_modules/torch_xla/distributed/spmd/xla_sharding.html index 625889b3be9..3c2b0ecd50a 100644 --- a/release/r2.4/_modules/torch_xla/distributed/spmd/xla_sharding.html +++ b/release/r2.4/_modules/torch_xla/distributed/spmd/xla_sharding.html @@ -265,7 +265,7 @@
- master (2.4.0+git18b351f ) + master (2.4.0+git8ac6425 )
diff --git a/release/r2.4/_modules/torch_xla/distributed/xla_multiprocessing.html b/release/r2.4/_modules/torch_xla/distributed/xla_multiprocessing.html index f1e4bfd981c..20701e0c386 100644 --- a/release/r2.4/_modules/torch_xla/distributed/xla_multiprocessing.html +++ b/release/r2.4/_modules/torch_xla/distributed/xla_multiprocessing.html @@ -265,7 +265,7 @@
- master (2.4.0+git18b351f ) + master (2.4.0+git8ac6425 )
diff --git a/release/r2.4/_modules/torch_xla/experimental/eager.html b/release/r2.4/_modules/torch_xla/experimental/eager.html index 4a16e07f14a..8c01994dc45 100644 --- a/release/r2.4/_modules/torch_xla/experimental/eager.html +++ b/release/r2.4/_modules/torch_xla/experimental/eager.html @@ -265,7 +265,7 @@
- master (2.4.0+git18b351f ) + master (2.4.0+git8ac6425 )
@@ -369,6 +369,7 @@

Source code for torch_xla.experimental.eager

 import functools
+from contextlib import contextmanager
 
 import torch_xla
 
@@ -388,6 +389,18 @@ 
Source code for torch_xla.experimental.eager
return torch_xla._XLAC._get_use_eager_mode()
 
 
+@contextmanager
+def eager_mode_context(enable: bool):
+  """Context manager to enable/disable the eager mode.
+  """
+  saved_eager_mode = is_eager_mode()
+  eager_mode(enable)
+  try:
+    yield saved_eager_mode
+  finally:
+    eager_mode(saved_eager_mode)
+
+
 
[docs]def compile(func):
   """Compile the func with Lazy Tensor.
 
@@ -405,12 +418,12 @@ 
Source code for torch_xla.experimental.eager
try:
       # Target Function Execution
       result = func(*args, **kwargs)
+      # Sync the graph generated by the target function.
+      torch_xla.sync()
     except Exception as e:
       # Handle exceptions (if needed)
       print(f"Error in target function: {e}")
       raise  # Re-raise the exception
-    # Sync the graph generated by the target function.
-    torch_xla.sync()
     torch_xla._XLAC._set_use_eager_mode(True)
 
     return result
diff --git a/release/r2.4/_modules/torch_xla/runtime.html b/release/r2.4/_modules/torch_xla/runtime.html
index 61265444cad..3b9e83f3d2f 100644
--- a/release/r2.4/_modules/torch_xla/runtime.html
+++ b/release/r2.4/_modules/torch_xla/runtime.html
@@ -265,7 +265,7 @@
               
               
                 

-                  master (2.4.0+git18b351f )
+                  master (2.4.0+git8ac6425 )
                 

               
             
diff --git a/release/r2.4/_modules/torch_xla/torch_xla.html b/release/r2.4/_modules/torch_xla/torch_xla.html
index 8c0ff4b11ec..6ebc39bbcf3 100644
--- a/release/r2.4/_modules/torch_xla/torch_xla.html
+++ b/release/r2.4/_modules/torch_xla/torch_xla.html
@@ -265,7 +265,7 @@
               
               
                 

-                  master (2.4.0+git18b351f )
+                  master (2.4.0+git8ac6425 )
                 

               
             
diff --git a/release/r2.4/genindex.html b/release/r2.4/genindex.html
index 0a30d3d7a9c..ca90b0021c3 100644
--- a/release/r2.4/genindex.html
+++ b/release/r2.4/genindex.html
@@ -265,7 +265,7 @@
               
               
                 

-                  master (2.4.0+git18b351f )
+                  master (2.4.0+git8ac6425 )
                 

               
             
diff --git a/release/r2.4/index.html b/release/r2.4/index.html
index 1b0c8b78343..49235b0f383 100644
--- a/release/r2.4/index.html
+++ b/release/r2.4/index.html
@@ -265,7 +265,7 @@
               
               
                 

-                  master (2.4.0+git18b351f )
+                  master (2.4.0+git8ac6425 )
                 

               
             
diff --git a/release/r2.4/notes/source_of_recompilation.html b/release/r2.4/notes/source_of_recompilation.html
index a52fef7db4a..683811a6e03 100644
--- a/release/r2.4/notes/source_of_recompilation.html
+++ b/release/r2.4/notes/source_of_recompilation.html
@@ -265,7 +265,7 @@
               
               
                 

-                  master (2.4.0+git18b351f )
+                  master (2.4.0+git8ac6425 )
                 

               
             
diff --git a/release/r2.4/objects.inv b/release/r2.4/objects.inv
index 5bcc62d2931aa51d22926ef6044aaa7fcaa5d31d..4fbf6430ff47b0f8ef97f559c27b1eecde9b2c01 100644
GIT binary patch
delta 18
ZcmaFP_MB}(0J}wEvYCmI>BevYW&l2}1*!l5

delta 18
ZcmaFP_MB}(0K1_@lCh~_+Qx7JW&k~$1*ZT2

diff --git a/release/r2.4/py-modindex.html b/release/r2.4/py-modindex.html
index 96a420b6d14..5bace7c060b 100644
--- a/release/r2.4/py-modindex.html
+++ b/release/r2.4/py-modindex.html
@@ -268,7 +268,7 @@
               
               
                 

-                  master (2.4.0+git18b351f )
+                  master (2.4.0+git8ac6425 )
                 

               
             
diff --git a/release/r2.4/search.html b/release/r2.4/search.html
index c40ae560d4e..ee66c4e02f0 100644
--- a/release/r2.4/search.html
+++ b/release/r2.4/search.html
@@ -265,7 +265,7 @@
               
               
                 

-                  master (2.4.0+git18b351f )
+                  master (2.4.0+git8ac6425 )