cudnn_utils.cpp 522 Bytes
Newer Older
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
/*************************************************************************
 * Copyright (c) 2022-2024, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 *
 * See LICENSE for license information.
 ************************************************************************/

#include "../fused_attn/utils.h"
#include "transformer_engine/cudnn.h"

namespace transformer_engine {

void nvte_cudnn_handle_init() {
  auto handle = cudnnExecutionPlanManager::Instance().GetCudnnHandle();
}

}  // namespace transformer_engine