Remove contiguity assertion from XLATensorImpl. (#7998)

pytorch · Sep 16, 2024 · 03374cd · 03374cd
1 parent 3b0097d
commit 03374cd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/torch_xla/csrc/tensor_impl.cpp b/torch_xla/csrc/tensor_impl.cpp
@@ -173,8 +173,8 @@ int64_t XLATensorImpl::numel_custom() const {
 }
 
 bool XLATensorImpl::is_contiguous_custom(at::MemoryFormat memory_format) const {
-  // Only check that the storage is already contiguous.
-  XLA_CHECK(is_contiguous_) << "Non-contiguous storage for XLA tensor";
+  // Storage is always contiguous, but the tensor metadata is_contiguous_ might
+  // be false due to the update in the functionalization layer..
   return true;
 }