Skip to content

GitLab

Unverified Commit 1be97cce authored by ngimel's avatar ngimel Committed by GitHub
Browse files

Merge pull request #190 from ngimel/fix_includes 

fix includes
parents 8f53411a f34686f1
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 3 deletions
csrc/multi_tensor_apply.cuh
View file @ 1be97cce
#include <torch/extension.h>
#include <ATen/ATen.h> #include <ATen/ATen.h>
#include <ATen/AccumulateType.h> #include <ATen/AccumulateType.h>
#include <ATen/cuda/CUDAContext.h> #include <ATen/cuda/CUDAContext.h>
#include <ATen/cuda/Exceptions.h> #include <ATen/cuda/Exceptions.h>
#include <assert.h> #include <assert.h>
#include <cuda_runtime.h>
// #include <iostream> // #include <iostream>
......
csrc/multi_tensor_scale_kernel.cu
View file @ 1be97cce
...@@ -5,7 +5,6 @@ ...@@ -5,7 +5,6 @@
#include "multi_tensor_apply.cuh" #include "multi_tensor_apply.cuh"
#include <assert.h> #include <assert.h>
#include <cuda_runtime.h>
#define BLOCK_SIZE 512 #define BLOCK_SIZE 512
#define ILP 4 #define ILP 4
......
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