1 #ifdef USE_CUDA 2 #include <ATen/cuda/CUDAConfig.h> // for the definition of AT_CUDNN_ENABLED 3 4 #if AT_CUDNN_ENABLED() 5 6 #include <ATen/ATen.h> 7 #include <ATen/native/quantized/cudnn/utils.h> 8 #include <ATen/native/quantized/PackedParams.h> 9 #include <torch/library.h> 10 11 #include <tuple> 12 unpack()13std::tuple<at::Tensor, std::optional<at::Tensor>> PackedLinearWeightCudnn::unpack() { 14 return std::tuple<at::Tensor, std::optional<at::Tensor>>{orig_weight, bias_}; 15 } 16 17 #endif // AT_CUDNN_ENABLED 18 #endif // USE_CUDA 19