Skip to content

GitLab

"src/vscode:/vscode.git/clone" did not exist on "bf9a641f1a51368af5f3ae99cc460107d4fa2103"
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 @@
#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 @ d6b5ae5d
......@@ -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