| 74 | } |
| 75 | |
| 76 | bool PoolingForwardImpl::AlgoCUDNN::is_available(const SizeArgs& args) const { |
| 77 | using Format = param::Pooling::Format; |
| 78 | return (((args.opr->param().format == Format::NCHW || |
| 79 | args.opr->param().format == Format::NHWC) && |
| 80 | (args.layout_src->dtype.enumv() == DTypeEnum::Float16 || |
| 81 | args.layout_src->dtype.enumv() == DTypeEnum::BFloat16 || |
| 82 | args.layout_src->dtype.enumv() == DTypeEnum::Float32 || |
| 83 | args.layout_src->dtype.enumv() == DTypeEnum::Int8 || |
| 84 | args.layout_src->dtype.enumv() == DTypeEnum::QuantizedS32 || |
| 85 | args.layout_src->dtype.enumv() == DTypeEnum::QuantizedS8 || |
| 86 | args.layout_src->dtype.enumv() == DTypeEnum::Quantized8Asymm)) || |
| 87 | ((args.opr->param().format == Format::NCHW4 || |
| 88 | args.opr->param().format == Format::NCHW32) && |
| 89 | (args.layout_src->dtype.enumv() == DTypeEnum::QuantizedS8 || |
| 90 | args.layout_src->dtype.enumv() == DTypeEnum::Quantized8Asymm))); |
| 91 | } |
| 92 | |
| 93 | void PoolingForwardImpl::AlgoCUDNN::init_mode( |
| 94 | const ExecArgs& args, cudnnPoolingMode_t& mode) const { |
no test coverage detected