chore: import upstream snapshot with attribution

2026-07-13 13:33:03 +08:00
commit 5b57521aa1
8226 changed files with 3425766 additions and 0 deletions
@@ -0,0 +1,34 @@
+//
+//  CastPlugin.cpp
+//  MNN
+//
+//  Created by MNN on b'2020/08/10'.
+//  Copyright © 2018, Alibaba Group Holding Limited
+//
+
+#include "CastPlugin.hpp"
+#include <MNN/MNNDefine.h>
+namespace MNN {
+
+CastPlugin::CastPlugin(const Op* op, const MNNTRTPlugin::Plugin* plugin) {
+    auto Info = plugin->main_as_OneHotInfo();
+    mCount   = Info->outerSize();
+}
+
+CastPlugin::~CastPlugin() {
+}
+
+int CastPlugin::onEnqueue(int batchSize, const void* const* inputs, void** outputs, void*, nvinfer1::DataType dataType, cudaStream_t stream) {
+
+    int size = 0;
+    if (dataType == nvinfer1::DataType::kFLOAT){
+        size = mCount*sizeof(float);
+    }else{
+        size = mCount*sizeof(__half);
+    }
+
+    auto status = cudaMemcpy(outputs[0], inputs[0], size, cudaMemcpyDeviceToDevice);
+    MNN_ASSERT(0 == status);
+}
+
+}; // namespace MNN