diff --git a/tools/pnnx/src/load_tnn.cpp b/tools/pnnx/src/load_tnn.cpp index 8b2fd6fd205..70b95a1d7f9 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"};