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