[LLGA] fix set_size of LlgaTensorImpl (#157)

* [LLGA] fix set_size

* [LLGA] do not support set_size for LlgaTensorImpl

* fix clang format

Author: chunyuan-w

torch_ipex/csrc/LlgaTensorImpl.cpp

@@ -13,10 +13,9 @@ LlgaTensorImpl::LlgaTensorImpl(
           c10::DispatchKeySet(DispatchKey::MkldnnCPU),
           data_type),
       desc_(desc) {
-        for (int64_t i = 0; i < desc.sizes().size(); i++) {
-          c10::TensorImpl::set_size(i, desc.sizes()[i]);
-        }
-      }
+  sizes_and_strides_.set_sizes(desc.sizes());
+  refresh_numel();
+}
 
 // The following are publically exposed as methods of Tensor
 IntArrayRef LlgaTensorImpl::strides() const {
@@ -35,6 +34,11 @@ int64_t LlgaTensorImpl::storage_offset() const {
   TORCH_CHECK(false, "Cannot access the storage_offset() of LlgaTensorImpl");
 }
 
+// The following are some internal inherited methods that we do not support.
+// They should never get called.
+void LlgaTensorImpl::set_size(int64_t dim, int64_t new_size) {
+  TORCH_INTERNAL_ASSERT(false, "Cannot set_size for LlgaTensorImpl");
+}
 void LlgaTensorImpl::set_stride(int64_t dim, int64_t new_stride) {
   TORCH_INTERNAL_ASSERT(false, "Cannot set_stride for LlgaTensorImpl");
 }

torch_ipex/csrc/LlgaTensorImpl.h

@@ -197,6 +197,7 @@ struct TORCH_API LlgaTensorImpl : public c10::TensorImpl {
           at::MemoryFormat::Contiguous) const override;
   IntArrayRef strides() const override;
   int64_t stride(int64_t d) const override;
+  void set_size(int64_t dim, int64_t new_size) override;
   void set_stride(int64_t dim, int64_t new_stride) override;
   void set_storage_offset(int64_t storage_offset) override;
   bool has_storage() const override;