Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
apex
Commits
cfe106d6
Commit
cfe106d6
authored
Jan 26, 2022
by
Jithun Nair
Browse files
Update ATen/CUDAGeneratorImpl.h to ATen/cuda/CUDAGeneratorImpl.h to resolve hipify issue
parent
1cb3da87
Changes
4
Hide whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
4 additions
and
4 deletions
+4
-4
apex/contrib/csrc/fmha/src/fmha.h
apex/contrib/csrc/fmha/src/fmha.h
+1
-1
apex/contrib/csrc/multihead_attn/dropout.h
apex/contrib/csrc/multihead_attn/dropout.h
+1
-1
apex/contrib/csrc/multihead_attn/softmax.h
apex/contrib/csrc/multihead_attn/softmax.h
+1
-1
apex/contrib/csrc/transducer/transducer_joint_kernel.cu
apex/contrib/csrc/transducer/transducer_joint_kernel.cu
+1
-1
No files found.
apex/contrib/csrc/fmha/src/fmha.h
View file @
cfe106d6
...
...
@@ -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>
...
...
apex/contrib/csrc/multihead_attn/dropout.h
View file @
cfe106d6
...
...
@@ -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>
...
...
apex/contrib/csrc/multihead_attn/softmax.h
View file @
cfe106d6
#pragma once
#include "philox.h"
#include <ATen/CUDAGeneratorImpl.h>
#include <ATen/
cuda/
CUDAGeneratorImpl.h>
#include <ATen/cuda/CUDAGraphsUtils.cuh>
#include <curand_kernel.h>
...
...
apex/contrib/csrc/transducer/transducer_joint_kernel.cu
View file @
cfe106d6
...
...
@@ -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>
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment