/**
 * \file dnn/src/naive/deformable_conv/opr_impl.h
 * MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
 *
 * Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
 *
 * Unless required by applicable law or agreed to in writing,
 * software distributed under the License is distributed on an
 * "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 */
#pragma once

#include "megdnn/oprs/nn.h"

namespace megdnn {
namespace naive {

class DeformableConvForwardImpl : public DeformableConvForward {
public:
    using DeformableConvForward::DeformableConvForward;

    std::vector<Algorithm*> get_all_algorithms(
            const TensorLayout& /* im */, const TensorLayout& /* filter */,
            const TensorLayout& /* offset */, const TensorLayout& /* mask */,
            const TensorLayout& /* dst */) override {
        return std::vector<Algorithm*>();
    };

    Algorithm* get_algorithm_heuristic(
            const TensorLayout& /* src */, const TensorLayout& /* filter */,
            const TensorLayout& /* offset */, const TensorLayout& /* mask */,
            const TensorLayout& /* dst */,
            size_t /* workspace_limit_in_bytes */,
            const AlgoAttribute& /*positive_attr*/,
            const AlgoAttribute& /*negative_attr*/) override {
        return nullptr;
    };

    size_t get_workspace_in_bytes(const TensorLayout& /* src */,
                                  const TensorLayout& /* filter */,
                                  const TensorLayout& /* offset */,
                                  const TensorLayout& /* mask */,
                                  const TensorLayout& /* dst */) override {
        return 0ULL;
    };

    const char* get_algorithm_set_name() const override {
        return "DEFORMABLE_CONV2_NAIVE";
    };

    Algorithm* get_algorithm_from_desc(const AlgorithmDesc&) override {
        return {};
    }

    void exec(_megdnn_tensor_in src, _megdnn_tensor_in filter,
              _megdnn_tensor_in offset, _megdnn_tensor_in mask,
              _megdnn_tensor_out dst, _megdnn_workspace workspace) override;
};

class DeformableConvBackwardFilterImpl : public DeformableConvBackwardFilter {
public:
    using DeformableConvBackwardFilter::DeformableConvBackwardFilter;

    std::vector<Algorithm*> get_all_algorithms(
            const TensorLayout& /* im */, const TensorLayout& /* offset */,
            const TensorLayout& /* mask */, const TensorLayout& /* out_grad */,
            const TensorLayout& /* filter_grad */) override {
        return std::vector<Algorithm*>();
    };

    Algorithm* get_algorithm_heuristic(
            const TensorLayout& /* im */, const TensorLayout& /* offset */,
            const TensorLayout& /* mask */, const TensorLayout& /* out_grad */,
            const TensorLayout& /* filter_grad */,
            size_t /* workspace_limit_in_bytes */,
            const AlgoAttribute& /*positive_attr*/,
            const AlgoAttribute& /*negative_attr*/) override {
        return nullptr;
    };

    size_t get_workspace_in_bytes(const TensorLayout& im,
                                  const TensorLayout& offset,
                                  const TensorLayout& mask,
                                  const TensorLayout& out_grad,
                                  const TensorLayout& filter_grad) override;

    const char* get_algorithm_set_name() const override {
        return "DEFORMABLE_CONV2_BWD_FILTER_NAIVE";
    };

    Algorithm* get_algorithm_from_desc(const AlgorithmDesc&) override {
        return {};
    }

    void exec(_megdnn_tensor_in im, _megdnn_tensor_in offset,
              _megdnn_tensor_in mask, _megdnn_tensor_in out_grad,
              _megdnn_tensor_out filter_grad,
              _megdnn_workspace workspace) override;
};

class DeformableConvBackwardDataImpl : public DeformableConvBackwardData {
public:
    using DeformableConvBackwardData::DeformableConvBackwardData;

    std::vector<Algorithm*> get_all_algorithms(
            const TensorLayout& /* im */, const TensorLayout& /* filter */,
            const TensorLayout& /* offset */, const TensorLayout& /* mask */,
            const TensorLayout& /* out_grad */,
            const TensorLayout& /* im_grad */,
            const TensorLayout& /* offset_grad */,
            const TensorLayout& /* mask_grad */) override {
        return std::vector<Algorithm*>();
    };

    Algorithm* get_algorithm_heuristic(
            const TensorLayout& /* im */, const TensorLayout& /* filter */,
            const TensorLayout& /* offset */, const TensorLayout& /* mask */,
            const TensorLayout& /* out_grad */,
            const TensorLayout& /* im_grad */,
            const TensorLayout& /* offset_grad */,
            const TensorLayout& /* mask_grad */,
            size_t /* workspace_limit_in_bytes */,
            const AlgoAttribute& /*positive_attr*/,
            const AlgoAttribute& /*negative_attr*/) override {
        return nullptr;
    };

    size_t get_workspace_in_bytes(const TensorLayout& im,
                                  const TensorLayout& filter,
                                  const TensorLayout& offset,
                                  const TensorLayout& mask,
                                  const TensorLayout& out_grad,
                                  const TensorLayout& im_grad,
                                  const TensorLayout& offset_grad,
                                  const TensorLayout& mask_grad) override;

    const char* get_algorithm_set_name() const override {
        return "DEFORMABLE_CONV2_BWD_DATA_NAIVE";
    };

    Algorithm* get_algorithm_from_desc(const AlgorithmDesc&) override {
        return {};
    }

    void exec(_megdnn_tensor_in im, _megdnn_tensor_in filter,
              _megdnn_tensor_in offset, _megdnn_tensor_in mask,
              _megdnn_tensor_in out_grad, _megdnn_tensor_out im_grad,
              _megdnn_tensor_out offset_grad, _megdnn_tensor_out mask_grad,
              _megdnn_workspace workspace) override;
};

}  // namespace naive
}  // namespace megdnn

// vim: syntax=cpp.doxygen