diff --git a/hpvm/lib/Transforms/HPVM2NVDLA/HPVM2NVDLAPass.cpp b/hpvm/lib/Transforms/HPVM2NVDLA/HPVM2NVDLAPass.cpp index e3b8f5dfc98eda3c9e60379693746849160a1ea8..300d55751832af46660cde6c02a81be5aa4d95de 100644 --- a/hpvm/lib/Transforms/HPVM2NVDLA/HPVM2NVDLAPass.cpp +++ b/hpvm/lib/Transforms/HPVM2NVDLA/HPVM2NVDLAPass.cpp @@ -20,6 +20,7 @@ #include <fstream> #include <vector> #include <map> +#include <set> #include "dlaerror.h" #include "dlatypes.h" @@ -1095,6 +1096,7 @@ void CGT_NVDLA::generateBatchNormLayer(DFLeafNode* N, const IntrinsicInst *II) { */ unsigned CGT_NVDLA::identifyOutputs() { + std::set< ITensor* > outputTensors; std::set< ITensor* > InputTensors;