Merge pull request #11 from NVIDIA/azrael417-patch-1

Update setup.cpp
NVIDIA · Nov 20, 2023 · 5c21c82 · 5c21c82
2 parents 8a697e5 + 64ea690
commit 5c21c82
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/csrc/setup.cpp b/src/csrc/setup.cpp
@@ -33,6 +33,7 @@
 #include <memory>
 #include <stdexcept>
 #include <string>
+#include <fstream>
 #include <unordered_map>
 
 #include <torch/torch.h>