From 5862671c0c9fcaebbdd8064c1de412197035e1ba Mon Sep 17 00:00:00 2001 From: Zhang Bolun Date: Mon, 6 May 2024 17:01:51 +0800 Subject: [PATCH] fix: add comments --- pyinfinitensor/src/pyinfinitensor/onnx.py | 1 + src/kernels/bang/resize.cc | 2 ++ 2 files changed, 3 insertions(+) diff --git a/pyinfinitensor/src/pyinfinitensor/onnx.py b/pyinfinitensor/src/pyinfinitensor/onnx.py index 91f6c976..0275f943 100644 --- a/pyinfinitensor/src/pyinfinitensor/onnx.py +++ b/pyinfinitensor/src/pyinfinitensor/onnx.py @@ -85,6 +85,7 @@ class OnnxStub: while len(sorted_nodes) < len(model.graph.node): updated = False for i, node in enumerate(model.graph.node): + # TODO：目前只考虑了resize算子输入为空的情况 if all(t in known_edge or t == "" for t in node.input): node.name = str(len(sorted_nodes)) + "_" + node.name sorted_nodes.append(i) diff --git a/src/kernels/bang/resize.cc b/src/kernels/bang/resize.cc index fbe47b9b..af44b2d3 100644 --- a/src/kernels/bang/resize.cc +++ b/src/kernels/bang/resize.cc @@ -85,6 +85,8 @@ class ResizeCnnl : public BangKernelWithoutConfig { cnnlCropAndResizeMode_t mode; auto coefMode = op->getMode(); if (coefMode == ResizeObj::ECoeffMode::nearest) { + // CNNL uses round by default and + // does not support other nearest modes mode = CNNL_CROP_AND_RESIZE_NEAREST; } else if (coefMode == ResizeObj::ECoeffMode::linear) { mode = CNNL_CROP_AND_RESIZE_BILINEAR;