From 72b78d5a4c3ab015a0f4d637fdd03af868a95933 Mon Sep 17 00:00:00 2001 From: nihui <171016+nihui@users.noreply.github.com> Date: Mon, 10 Feb 2025 08:56:24 +0000 Subject: [PATCH] apply code-format changes --- tools/pnnx/src/load_tnn.cpp | 10 +++++----- tools/pnnx/src/pass_level2/torch_index_select.cpp | 3 +-- 2 files changed, 6 insertions(+), 7 deletions(-) diff --git a/tools/pnnx/src/load_tnn.cpp b/tools/pnnx/src/load_tnn.cpp index aaf2950d733..634abc1c87c 100644 --- a/tools/pnnx/src/load_tnn.cpp +++ b/tools/pnnx/src/load_tnn.cpp @@ -139,11 +139,11 @@ static size_t type_to_elemsize(int type) static int get_tnn_tensor_type(int dt) { - if (dt == 0) return 1;// fp32 - if (dt == 1) return 3;// fp16 - if (dt == 2) return 7;// int8 - if (dt == 3) return 4;// int32 - if (dt == 4) return 13;// bf16 + if (dt == 0) return 1; // fp32 + if (dt == 1) return 3; // fp16 + if (dt == 2) return 7; // int8 + if (dt == 3) return 4; // int32 + if (dt == 4) return 13; // bf16 fprintf(stderr, "unsupported tnn tensor type %d\n", dt); return 0; // unknown type diff --git a/tools/pnnx/src/pass_level2/torch_index_select.cpp b/tools/pnnx/src/pass_level2/torch_index_select.cpp index f9fe8eec67d..2bfc10558d4 100644 --- a/tools/pnnx/src/pass_level2/torch_index_select.cpp +++ b/tools/pnnx/src/pass_level2/torch_index_select.cpp @@ -84,12 +84,11 @@ pnnx.Output output 1 0 out { indices[i] = p[i]; } - op_index->attrs["data"].type = 5;// i64 + op_index->attrs["data"].type = 5; // i64 op_index->attrs["data"].shape = {n}; op_index->attrs["data"].data.resize(n * 8); memcpy((void*)op_index->attrs["data"].data.data(), (const void*)indices.data(), n * 8); - Operator* op_gather = ops.at("select"); op_gather->params["dim"] = captured_params.at("dim"); op_gather->inputnames = {"input", "index"};