Skip to content

GitLab

"git@developer.sourcefind.cn:OpenDAS/ktransformers.git" did not exist on "57cf449a97097045b560001d1dcb9ce5754be42f"
Unverified Commit d6b5ae5d authored by X Wang's avatar X Wang Committed by GitHub
Browse files

include iostream (#1144)

parent 0c2c6eea
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
apex/contrib/csrc/groupbn/batch_norm.h
View file @ d6b5ae5d
...@@ -31,6 +31,7 @@ ...@@ -31,6 +31,7 @@
#include <algorithm> #include <algorithm>
#include <vector> #include <vector>
#include <string> #include <string>
#include <iostream>
#include "nhwc_batch_norm_kernel.h" #include "nhwc_batch_norm_kernel.h"
#include "cuda_utils.h" #include "cuda_utils.h"
......
apex/contrib/csrc/groupbn/batch_norm_add_relu.h
View file @ d6b5ae5d
...@@ -31,6 +31,7 @@ ...@@ -31,6 +31,7 @@
#include <algorithm> #include <algorithm>
#include <vector> #include <vector>
#include <string> #include <string>
#include <iostream>
#include "nhwc_batch_norm_kernel.h" #include "nhwc_batch_norm_kernel.h"
#include "cuda_utils.h" #include "cuda_utils.h"
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment