Commit cfe106d6 authored by Jithun Nair's avatar Jithun Nair
Browse files

Update ATen/CUDAGeneratorImpl.h to ATen/cuda/CUDAGeneratorImpl.h to resolve hipify issue

parent 1cb3da87
...@@ -30,7 +30,7 @@ ...@@ -30,7 +30,7 @@
#include <cuda.h> #include <cuda.h>
#include <vector> #include <vector>
#include <ATen/CUDAGeneratorImpl.h> #include <ATen/cuda/CUDAGeneratorImpl.h>
#include <ATen/cuda/CUDAGraphsUtils.cuh> #include <ATen/cuda/CUDAGraphsUtils.cuh>
#include <fmha_utils.h> #include <fmha_utils.h>
......
...@@ -3,7 +3,7 @@ ...@@ -3,7 +3,7 @@
#ifdef OLD_GENERATOR #ifdef OLD_GENERATOR
#include <ATen/CUDAGenerator.h> #include <ATen/CUDAGenerator.h>
#else #else
#include <ATen/CUDAGeneratorImpl.h> #include <ATen/cuda/CUDAGeneratorImpl.h>
#endif #endif
#include <ATen/cuda/CUDAContext.h> #include <ATen/cuda/CUDAContext.h>
......
#pragma once #pragma once
#include "philox.h" #include "philox.h"
#include <ATen/CUDAGeneratorImpl.h> #include <ATen/cuda/CUDAGeneratorImpl.h>
#include <ATen/cuda/CUDAGraphsUtils.cuh> #include <ATen/cuda/CUDAGraphsUtils.cuh>
#include <curand_kernel.h> #include <curand_kernel.h>
......
...@@ -4,7 +4,7 @@ ...@@ -4,7 +4,7 @@
#include <torch/extension.h> #include <torch/extension.h>
#include <ATen/AccumulateType.h> #include <ATen/AccumulateType.h>
#include <ATen/CUDAGeneratorImpl.h> #include <ATen/cuda/CUDAGeneratorImpl.h>
#include <ATen/cuda/CUDAContext.h> #include <ATen/cuda/CUDAContext.h>
#include <ATen/cuda/CUDAGraphsUtils.cuh> #include <ATen/cuda/CUDAGraphsUtils.cuh>
#include <c10/macros/Macros.h> #include <c10/macros/Macros.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