Skip to content

GitLab

Commit 51b402df authored by X Wang's avatar X Wang Committed by Hubert Lu
Browse files

include iostream (#1144)

parent abb6e5ba
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
apex/contrib/csrc/groupbn/batch_norm.h
View file @ 51b402df
...@@ -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 @ 51b402df
...@@ -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