Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "c0933457afb79e250d3f713da1d7f359cdea5576"
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 @@
#include <algorithm>
#include <vector>
#include <string>
#include <iostream>
#include "nhwc_batch_norm_kernel.h"
#include "cuda_utils.h"
......
apex/contrib/csrc/groupbn/batch_norm_add_relu.h
View file @ 51b402df
......@@ -31,6 +31,7 @@
#include <algorithm>
#include <vector>
#include <string>
#include <iostream>
#include "nhwc_batch_norm_kernel.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