Skip to content

GitLab

Commit f34686f1 authored by Natalia Gimelshein's avatar Natalia Gimelshein
Browse files

fix includes

parent 8f53411a
Hide whitespace changes
Inline Side-by-side
Showing with 0 additions and 3 deletions
csrc/multi_tensor_apply.cuh
View file @ f34686f1
#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 @ f34686f1
...@@ -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