Skip to content

GitLab

Commit 76ee6412 authored by zhuww's avatar zhuww
Browse files

modify include <cooperative_groups.h>

parent 8e85aa97
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 1 deletion
fastfold/model/fastnn/kernel/cuda_native/csrc/layer_norm_cuda_kernel.cu
View file @ 76ee6412
// part of code modified from https://github.com/NVIDIA/apex // part of code modified from https://github.com/NVIDIA/apex
#include <cooperative_groups.h> //#include <cooperative_groups.h>
#include <hip/hsa_detail/hip_cooperative_groups.h>
#include <cuda.h> #include <cuda.h>
#include <cuda_runtime.h> #include <cuda_runtime.h>
#include <torch/extension.h> #include <torch/extension.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