update graph.py

DeepLink-org · Jan 26, 2024 · bfdad45 · bfdad45
1 parent 837f289
commit bfdad45
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/dicp/dicp/dynamo_bridge/graph.py b/dicp/dicp/dynamo_bridge/graph.py
@@ -65,10 +65,11 @@ def make_tensor_meta(x) -> Optional[TensorMetadata]:
                 for i,(meta_i,fv_i) in enumerate(zip(n_meta_val, fake_val)):
                     if not isinstance(fv_i, FakeTensor):
                         continue
-                    assert meta_i.size() == fv_i.size(), "check infer size failed"
-                    assert meta_i.dtype == fv_i.dtype, "check infer dtype failed"
-                    assert meta_i.stride() == fv_i.stride(), "check infer stride failed"
-                    assert meta_i.storage_offset() == fv_i.storage_offset(), "check infer storage offset failed"
+                    tensor_info = f"meta_i: {meta_i}, fv_i: {fv_i}"
+                    assert meta_i.size() == fv_i.size(), f"check infer size failed, {tensor_info}"
+                    assert meta_i.dtype == fv_i.dtype, f"check infer dtype failed, {tensor_info}"
+                    assert meta_i.stride() == fv_i.stride(), f"check infer stride failed, {tensor_info}"
+                    assert meta_i.storage_offset() == fv_i.storage_offset(), f"check infer storage offset failed, {tensor_info}"
             if 'val' not in n.meta:
                 n.meta['val'] = fake_value
                 n.meta["tensor_meta"] = make_tensor_meta(n.meta['val'])