cranky-coder08
/

Phi2-Fine-Tuning

Add files using upload-large-folder tool

c1af2fa verified 3 months ago

202 Bytes

	#pragma once

	#include <ATen/cuda/ATenCUDAGeneral.h>
	#include <ATen/cudnn/cudnn-wrapper.h>

	namespace at::native {

	TORCH_CUDA_CPP_API cudnnHandle_t getCudnnHandle();
	} // namespace at::native