Skip to content

Commit 5a34bc1

Browse files
authored
Qualcomm AI Engine Direct - Fix Per Layer Dump (#6384)
1 parent 484774f commit 5a34bc1

File tree

1 file changed

+5
-3
lines changed

1 file changed

+5
-3
lines changed

backends/qualcomm/builders/node_visitor.py

Lines changed: 5 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -228,9 +228,11 @@ def get_tensor_type(
228228

229229
if is_parameter(node, self.edge_program):
230230
return PyQnnWrapper.Qnn_TensorType_t.QNN_TENSOR_TYPE_STATIC
231-
232-
# dump all tensor, set to app read
233-
if self.enable_tensor_dump:
231+
# dump all tensor, set to app read, and we only dump native tensors
232+
if (
233+
self.enable_tensor_dump
234+
and tensor_type == PyQnnWrapper.Qnn_TensorType_t.QNN_TENSOR_TYPE_NATIVE
235+
):
234236
return PyQnnWrapper.Qnn_TensorType_t.QNN_TENSOR_TYPE_APP_READ
235237
return tensor_type
236238

0 commit comments

Comments
 (0)