[examples/python/pytorch] added skeleton of wrapper for shift-conv and batch-norm

2019-07-09 21:54:37 -07:00
parent 63b249c1d6
commit 3b89bc8463
8 changed files with 63 additions and 151 deletions
--- a/examples/python/pytorch/batchnorm.cpp
+++ b/examples/python/pytorch/batchnorm.cpp
@@ -3,7 +3,6 @@
 #include "ATen/cuda/CUDAContext.h"
 #include "triton/driver/stream.h"
 #include "triton/dnn/batchnorm.h"
-#include "triton/tools/bench.hpp"

 std::vector<torch::Tensor>
      batchnorm_ymv(const torch::Tensor fw_x,