* Copyright (c) 2026 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 vf_ops.cpp
* \brief VF API IR op registration.
*
* Registers VF ops with type deduction for the direct VF instruction path.
* These ops are part of the merged shared IR surface and remain consumed by
* the block VF frontend/backend path.
*/
#include <any>
#include <string>
#include <utility>
#include <vector>
#include "core/logging.h"
#include "ir/kind_traits.h"
#include "ir/op_registry.h"
#include "ir/type.h"
namespace pypto {
namespace ir {
namespace {
TypePtr DeduceVFUnknownType(
[[maybe_unused]] const std::vector<ExprPtr>& args,
[[maybe_unused]] const std::vector<std::pair<std::string, std::any>>& kwargs)
{
return GetUnknownType();
}
TypePtr DeduceVFScalarType(
[[maybe_unused]] const std::vector<ExprPtr>& args,
[[maybe_unused]] const std::vector<std::pair<std::string, std::any>>& kwargs)
{
return std::make_shared<ScalarType>(DataType::FP32);
}
TypePtr DeduceVFMaskType(
[[maybe_unused]] const std::vector<ExprPtr>& args,
[[maybe_unused]] const std::vector<std::pair<std::string, std::any>>& kwargs)
{
return std::make_shared<ScalarType>(DataType::UINT16);
}
TypePtr DeduceVFFromFirstArg(
[[maybe_unused]] const std::vector<ExprPtr>& args,
[[maybe_unused]] const std::vector<std::pair<std::string, std::any>>& kwargs)
{
if (!args.empty()) {
return args[0]->GetType();
}
return std::make_shared<ScalarType>(DataType::FP32);
}
TypePtr DeduceVFFromSecondArg(
[[maybe_unused]] const std::vector<ExprPtr>& args,
[[maybe_unused]] const std::vector<std::pair<std::string, std::any>>& kwargs)
{
if (args.size() >= 0x2) {
return args[1]->GetType();
}
return std::make_shared<ScalarType>(DataType::FP32);
}
TypePtr DeduceVFFromDstArg(
[[maybe_unused]] const std::vector<ExprPtr>& args,
[[maybe_unused]] const std::vector<std::pair<std::string, std::any>>& kwargs)
{
if (args.size() >= 0x3) {
return args[2]->GetType();
}
return std::make_shared<ScalarType>(DataType::FP32);
}
}
REGISTER_OP("vf.vf_scope_enter")
.set_op_category("VFOp")
.set_description("VF scope enter marker")
.no_argument()
.f_deduce_type(DeduceVFUnknownType);
REGISTER_OP("vf.vf_scope_exit")
.set_op_category("VFOp")
.set_description("VF scope exit marker")
.no_argument()
.f_deduce_type(DeduceVFUnknownType);
REGISTER_OP("vf.RegTensor")
.set_op_category("VFOp")
.set_description("Declare a vector register")
.no_argument()
.f_deduce_type(DeduceVFScalarType);
REGISTER_OP("vf.CreateMask")
.set_op_category("VFOp")
.set_description("Declare and initialize a mask register")
.no_argument()
.f_deduce_type(DeduceVFMaskType);
REGISTER_OP("vf.Duplicate")
.set_op_category("VFOp")
.set_description("Scalar broadcast to register")
.add_argument("dst", "Destination register")
.f_deduce_type(DeduceVFFromFirstArg);
REGISTER_OP("vf.LoadAlign")
.set_op_category("VFOp")
.set_description("Load aligned data from UB to register")
.add_argument("dst", "Destination register")
.f_deduce_type(DeduceVFFromFirstArg);
REGISTER_OP("vf.StoreAlign")
.set_op_category("VFOp")
.set_description("Store aligned data from register to UB")
.add_argument("src", "Source register")
.add_argument("mask", "Mask register")
.f_deduce_type(DeduceVFUnknownType);
REGISTER_OP("vf.Max")
.set_op_category("VFOp")
.set_description("Vector maximum")
.add_argument("src0", "Source register 0")
.add_argument("src1", "Source register 1")
.add_argument("dst", "Destination register")
.add_argument("mask", "Mask register")
.f_deduce_type(DeduceVFFromDstArg);
REGISTER_OP("vf.Add")
.set_op_category("VFOp")
.set_description("Vector addition")
.add_argument("src0", "Source register 0")
.add_argument("src1", "Source register 1")
.add_argument("dst", "Destination register")
.add_argument("mask", "Mask register")
.f_deduce_type(DeduceVFFromDstArg);
REGISTER_OP("vf.Sub")
.set_op_category("VFOp")
.set_description("Vector subtraction")
.add_argument("src0", "Source register 0")
.add_argument("src1", "Source register 1")
.add_argument("dst", "Destination register")
.add_argument("mask", "Mask register")
.f_deduce_type(DeduceVFFromDstArg);
REGISTER_OP("vf.Mul")
.set_op_category("VFOp")
.set_description("Vector multiplication")
.add_argument("src0", "Source register 0")
.add_argument("src1", "Source register 1")
.add_argument("dst", "Destination register")
.add_argument("mask", "Mask register")
.f_deduce_type(DeduceVFFromDstArg);
REGISTER_OP("vf.Div")
.set_op_category("VFOp")
.set_description("Vector division")
.add_argument("src0", "Source register 0")
.add_argument("src1", "Source register 1")
.add_argument("dst", "Destination register")
.add_argument("mask", "Mask register")
.f_deduce_type(DeduceVFFromDstArg);
REGISTER_OP("vf.Muls")
.set_op_category("VFOp")
.set_description("Scalar multiplication")
.add_argument("src", "Source register")
.add_argument("dst", "Destination register")
.add_argument("mask", "Mask register")
.f_deduce_type(DeduceVFFromSecondArg);
REGISTER_OP("vf.Ln")
.set_op_category("VFOp")
.set_description("Natural logarithm")
.add_argument("src", "Source register")
.add_argument("dst", "Destination register")
.add_argument("mask", "Mask register")
.f_deduce_type(DeduceVFFromSecondArg);
REGISTER_OP("vf.FusedExpSub")
.set_op_category("VFOp")
.set_description("Fused exp(src - max)")
.add_argument("src", "Source register")
.add_argument("max", "Max register")
.add_argument("dst", "Destination register")
.add_argument("mask", "Mask register")
.f_deduce_type(DeduceVFFromDstArg);
REGISTER_OP("vf.Cast")
.set_op_category("VFOp")
.set_description("Type conversion")
.add_argument("src", "Source register")
.add_argument("dst", "Destination register")
.add_argument("mask", "Mask register")
.f_deduce_type(DeduceVFFromSecondArg);
REGISTER_OP("vf.DeInterleave")
.set_op_category("VFOp")
.set_description("De-interleave")
.add_argument("src0", "Source register 0")
.add_argument("src1", "Source register 1")
.add_argument("dst0", "Destination register 0")
.add_argument("dst1", "Destination register 1")
.f_deduce_type(DeduceVFFromDstArg);
REGISTER_OP("vf.Select")
.set_op_category("VFOp")
.set_description("Conditional select")
.add_argument("src_true", "True branch register")
.add_argument("src_false", "False branch register")
.add_argument("dst", "Destination register")
.add_argument("mask", "Mask register")
.f_deduce_type(DeduceVFFromDstArg);
REGISTER_OP("vf.UpdateMask")
.set_op_category("VFOp")
.set_description("Update mask with scalar value")
.add_argument("scalar", "Scalar value for mask")
.f_deduce_type(DeduceVFMaskType);
REGISTER_OP("vf.MemBar")
.set_op_category("VFOp")
.set_description("Memory barrier")
.no_argument()
.f_deduce_type(DeduceVFUnknownType);
}
}
