diff --git a/paddle/cuda/include/hl_base.h b/paddle/cuda/include/hl_base.h index b979aa7723ea77fcae98a5be41d9b9dc488a68c3..402302a5bfa877be33bf3bbd6ba05c8f3e48c3ba 100644 --- a/paddle/cuda/include/hl_base.h +++ b/paddle/cuda/include/hl_base.h @@ -12,8 +12,7 @@ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the License for the specific language governing permissions and limitations under the License. */ -#ifndef HL_BASE_H_ -#define HL_BASE_H_ +#pragma once #include @@ -207,8 +206,8 @@ typedef struct { #ifdef __NVCC__ -#include "cuda_runtime.h" -#include "hl_cuda.h" +#include +#include "paddle/cuda/include/hl_cuda.h" #include "paddle/utils/Logging.h" extern __thread bool g_sync_flag; @@ -243,6 +242,4 @@ __shfl_sync(unsigned, T val, int src_line, int width) { mask = __ballot_sync(FULL_WARP_MASK, (predicate)) #endif -#endif /* __NVCC__ */ - -#endif /* HL_BASE_H_ */ +#endif // __NVCC__