[easy] Clean up temporary files

MatzeB · Nov 8, 2018 · b44b9a7 · b44b9a7
1 parent e8d9562
commit b44b9a7
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/lib/Graph/Graph.cpp b/lib/Graph/Graph.cpp
@@ -250,6 +250,7 @@ void Module::dumpDAG() {
   llvm::SmallString<64> dotPath;
   llvm::sys::fs::createTemporaryFile("dotty_graph_dump", "dot", dotPath);
   dumpDAG(dotPath);
+  llvm::sys::fs::remove(dotPath);
 }
 
 void Module::dumpDAG(llvm::StringRef dotFilename) {
@@ -2177,6 +2178,7 @@ void Function::dumpDAG() {
   llvm::SmallString<64> dotPath;
   llvm::sys::fs::createTemporaryFile("dotty_graph_dump", "dot", dotPath);
   dumpDAG(dotPath);
+  llvm::sys::fs::remove(dotPath);
 }
 
 void Function::dumpDAG(llvm::StringRef dotFilename) {

diff --git a/tests/unittests/quantizationTest.cpp b/tests/unittests/quantizationTest.cpp
@@ -57,7 +57,7 @@ void testSerialization(const std::vector<NodeQuantizationInfo> &expected) {
   serializeToYaml(filePath, expected);
   std::vector<NodeQuantizationInfo> deserialized =
       deserializeFromYaml(filePath);
-
+  llvm::sys::fs::remove(filePath);
   EXPECT_EQ(expected, deserialized);
 }