/**

 * Copyright (c) 2025 Huawei Technologies Co., Ltd.

 * This program is free software, you can redistribute it and/or modify it under the terms and conditions of

 * CANN Open Software License Agreement Version 2.0 (the "License").

 * Please refer to the License for details. You may not use this file except in compliance with the License.

 * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED,

 * INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY, OR FITNESS FOR A PARTICULAR PURPOSE.

 * See LICENSE in the root of the software repository for the full text of the License.

 */



/*!

 * \file process_atomic_checker.h

 * \brief Checker for ProcessAtomic pass

 */



#ifndef PROCESS_ATOMIC_CHECKER_H

#define PROCESS_ATOMIC_CHECKER_H



#include "checker.h"

#include "interface/operation/opcode.h"

#include "interface/function/function.h"

#include "interface/operation/operation.h"

#include "interface/tensor/logical_tensor.h"



namespace npu {

namespace tile_fwk {

class ProcessAtomicChecker : Checker {

public:

    Status DoPreCheck(Function& function) override;

    Status DoPostCheck(Function& function) override;



private:

    Status CheckMulOpValidity(const Operation& op);

    Status CheckReduceAccOpValidity(const Operation& op);

    Status CheckAtomicRMWMemoryType(const Operation& op);

    Status CheckAtomicRMWShape(const Operation& op);

    Status CheckAtomicRMWOffset(const Operation& op);

    Status ValidateAtomicRMW(const Operation& op);

    Status ProcessPreCheck(const Operation& op);

    Status CheckNoReduceAcc(Function& function);

    Status CheckNoAtomicRMW(Function& function);

};

} // namespace tile_fwk

} // namespace npu

#endif // PROCESS_ATOMIC_CHECKER_H