Change LOD_TENSOR to DENSE_TENSOR

Author: co63oc

paddleslim/common/recover_program.py

@@ -45,7 +45,7 @@ def _recover_outputs_attr(program):
                         unique_name.generate_with_ignorable_key(
                             ".".join(["reserve_space", 'tmp'])),
                         dtype=block.var(op.input("X")[0]).dtype,
-                        type=paddle.framework.core.VarDesc.VarType.LOD_TENSOR,
+                        type=paddle.framework.core.VarDesc.VarType.DENSE_TENSOR,
                         persistable=False,
                         stop_gradient=True)
                     op.desc.set_output("ReserveSpace", [reserve_space.name])
@@ -56,7 +56,7 @@ def _recover_outputs_attr(program):
                         unique_name.generate_with_ignorable_key(
                             ".".join(["xshape", 'tmp'])),
                         dtype=block.var(op.input("X")[0]).dtype,
-                        type=paddle.framework.core.VarDesc.VarType.LOD_TENSOR,
+                        type=paddle.framework.core.VarDesc.VarType.DENSE_TENSOR,
                         shape=(0, ) + block.var(op.input("X")[0]).shape,
                         persistable=False,
                         stop_gradient=True)
@@ -67,7 +67,7 @@ def _recover_outputs_attr(program):
                         name=paddle.utils.unique_name.
                         generate_with_ignorable_key(".".join(["mask", 'tmp'])),
                         dtype=block.var(op.input("X")[0]).dtype,
-                        type=paddle.framework.core.VarDesc.VarType.LOD_TENSOR,
+                        type=paddle.framework.core.VarDesc.VarType.DENSE_TENSOR,
                         shape=block.var(op.input("X")[0]).shape,
                         persistable=False,
                         stop_gradient=True)

paddleslim/quant/quant_embedding.py

@@ -397,7 +397,7 @@ def _split_embedding_seq_pool(graph, op):
     out = outputs[0]
     lookup_out = graph.create_var_node(
         name=ids.name() + '.look_up_table.out',
-        var_type=paddle.framework.core.VarDesc.VarType.LOD_TENSOR,
+        var_type=paddle.framework.core.VarDesc.VarType.DENSE_TENSOR,
         shape=[1],
         var_dtype=weight.dtype())
     lookup_table_op = graph.create_op_node(
@@ -415,7 +415,7 @@ def _split_embedding_seq_pool(graph, op):
     graph.link_to(lookup_table_op, lookup_out)
     max_index = graph.create_var_node(
         name=ids.name() + '.seq_pool_op.max_index',
-        var_type=paddle.framework.core.VarDesc.VarType.LOD_TENSOR,
+        var_type=paddle.framework.core.VarDesc.VarType.DENSE_TENSOR,
         shape=[1],
         var_dtype=weight.dtype())
 

paddleslim/quant/quanters/channel_wise_abs_max.py

@@ -136,7 +136,7 @@ def static_forward(self, input):
         quant_out = self._helper.create_variable(
             name=f"{input.name}.quantized.dequantized",
             dtype=input.dtype,
-            type=core.VarDesc.VarType.LOD_TENSOR,
+            type=core.VarDesc.VarType.DENSE_TENSOR,
             persistable=False,
             stop_gradient=False, )
         outputs = {"Out": [quant_out], "OutScale": [self._scale]}