squall/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

add #pragma once for header file in example 42 (#698)

Browse Source
This commit is contained in:
Wenzhuo Liu 2022-11-16 11:50:24 +08:00 committed by GitHub
parent 06eb90cc0d
commit 3c90f6aea6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
examples/42_fused_multi_head_attention/kernel_forward.h
View File

@ -1,3 +1,5 @@
#pragma once
#ifdef HAS_PYTORCH
#include <ATen/ATen.h>
#include <ATen/cuda/CUDAContext.h>