fix debug only assert

Author: chunyuanw

tests/cpu/test_lazy_reorder.py

@@ -153,6 +153,7 @@ def test_seq_deconv(self):
 
 class TestBinaryOp(TestCase):
     def test_add(self):
+        # rand_seed = 1599794793172034560: AssertionError: tensor(1.5259e-05) not less than or equal to 1e-05
         ipex.core.enable_auto_dnnl()
         rand_seed = int(get_rand_seed())
         print("{} rand sed: {}".format(sys._getframe().f_code.co_name, rand_seed))

torch_ipex/csrc/cpu/ShadeDataContext.h

@@ -201,12 +201,11 @@ struct ShadeDataContext {
   static inline bool isTensorMixPrecision(const at::Tensor &tensor) {
     auto dil_tensor_type = getDilStorage(tensor).get_data_type();
     auto aten_tensor_type = tensor.scalar_type();
-    TORCH_INTERNAL_ASSERT_DEBUG_ONLY(aten_tensor_type == at::kFloat || aten_tensor_type == at::kBFloat16
-        || aten_tensor_type == at::kByte || aten_tensor_type == at::kChar);
-    TORCH_INTERNAL_ASSERT_DEBUG_ONLY(dil_tensor_type == dil::data_type::bf16 || dil_tensor_type == dil::data_type::f32
+    if (aten_tensor_type != at::kFloat) {
+      return false;
+    }
+    auto res = (dil_tensor_type == dil::data_type::bf16 
         || dil_tensor_type == dil::data_type::u8 || dil_tensor_type == dil::data_type::s8);
-    auto res = (dil_tensor_type == dil::data_type::bf16 || dil_tensor_type == dil::data_type::u8
-        || dil_tensor_type == dil::data_type::s8) && aten_tensor_type == at::kFloat ;
 
     // Check mix_precision
     void *raw_context = tensor.storage().data_ptr().get_context();

torch_ipex/csrc/cpu/dbl/Common.cpp

@@ -296,8 +296,7 @@ void reorder_to_public(const at::Tensor& tensor, bool remain_dtype) {
   bool is_public_format = dil_buffer.is_public_format();
   bool is_public_dtype = !cpu::ShadeDataContext::isTensorMixPrecision(tensor);
 
-  TORCH_INTERNAL_ASSERT_DEBUG_ONLY(
-      is_public_dtype && (aten_dtype == at::kFloat || aten_dtype == at::kBFloat16) || !is_public_dtype && aten_dtype == at::kFloat)
+  TORCH_INTERNAL_ASSERT_DEBUG_ONLY(is_public_dtype || (!is_public_dtype && aten_dtype == at::kFloat));
 
   bool should_reorder_format = !is_public_format;
   bool should_reorder_dtype = !is_public_dtype && !remain_dtype;