Skip to content

GitLab

Commit 2228f1bf authored by Hubert Lu's avatar Hubert Lu
Browse files

Merge remote-tracking branch 'origin/master' into dev/hubertlu/unit_tests

parents 3f3da214 08e88b1b
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
apex/contrib/csrc/groupbn/batch_norm.h
View file @ 2228f1bf
...@@ -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 @ 2228f1bf
...@@ -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