tvm
Loading...
Searching...
No Matches
Namespaces | Macros | Functions
ir.h File Reference
#include <tvm/ffi/container/tuple.h>
#include <tvm/ffi/container/variant.h>
#include <tvm/ir/prim/op.h>
#include <tvm/runtime/tensor.h>
#include <tvm/script/ir_builder/ir.h>
#include <tvm/tirx/script/ir_builder/frame.h>
#include <tvm/tirx/stmt.h>
#include <tvm/tirx/tile/exec_scope.h>
#include <tvm/tirx/type.h>
Include dependency graph for ir.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces

namespace  tvm
 An object that builds and maintains block scope and StmtSref mapping for Dependence analysis.
 
namespace  tvm::script
 
namespace  tvm::script::ir_builder
 
namespace  tvm::script::ir_builder::tirx
 

Macros

#define TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST(FuncName, DType)
 
#define TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST_SIZES(DType, Code)
 
#define TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST_LANES(FuncName, Code, Size)
 
#define TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST_SIZES_LANES(DType, Code)
 
#define TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST_LANES_FIXED_SIZE(DType, Code, Bits)
 

Functions

FunctionFrame tvm::script::ir_builder::tirx::Function (bool is_private, bool persistent=false)
 The primitive function statement.
 
FunctionFrame tvm::script::ir_builder::tirx::DeclFunction (bool is_private=false, bool persistent=false)
 Construct a bodyless function signature using the ordinary signature operations.
 
Var tvm::script::ir_builder::tirx::Arg (ffi::String name, Var var)
 The Function variable arguments adding function.
 
TensorVar tvm::script::ir_builder::tirx::Arg (ffi::String name, TensorVar tensor)
 The Function tensor arguments adding function.
 
void tvm::script::ir_builder::tirx::FuncName (ffi::String name)
 The Function naming statement.
 
void tvm::script::ir_builder::tirx::FuncAttrs (ffi::Map< ffi::String, ffi::Any > attrs)
 The Function annotation statement.
 
Type tvm::script::ir_builder::tirx::FuncRet (Type ret_type)
 The Function return type statement.
 
tvm::Stmt tvm::script::ir_builder::tirx::TensorStore (Expr dest, ffi::Array< PrimExpr > indices, PrimExpr value)
 Store data in a tensor destination.
 
Var tvm::script::ir_builder::tirx::Handle (ffi::Optional< PrimType > dtype=std::nullopt, ffi::String storage_scope="global")
 Create a TIR var that represents a pointer.
 
Var tvm::script::ir_builder::tirx::TensorMap ()
 
PrimExpr tvm::script::ir_builder::tirx::BFloat8 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::BFloat16 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::BFloat32 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::BFloat64 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float16 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float32 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float64 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::UInt8 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::UInt16 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::UInt32 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::UInt64 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Int8 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Int16 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Int32 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Int64 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::BFloat8x2 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::BFloat8x4 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::BFloat8x8 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::BFloat8x16 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::BFloat8x32 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::BFloat8x64 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::BFloat16x2 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::BFloat16x4 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::BFloat16x8 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::BFloat16x16 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::BFloat16x32 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::BFloat16x64 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::BFloat32x2 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::BFloat32x4 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::BFloat32x8 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::BFloat32x16 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::BFloat32x32 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::BFloat32x64 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::BFloat64x2 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::BFloat64x4 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::BFloat64x8 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::BFloat64x16 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::BFloat64x32 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::BFloat64x64 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8x2 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8x4 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8x8 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8x16 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8x32 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8x64 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float16x2 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float16x4 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float16x8 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float16x16 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float16x32 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float16x64 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float32x2 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float32x4 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float32x8 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float32x16 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float32x32 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float32x64 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float64x2 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float64x4 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float64x8 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float64x16 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float64x32 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float64x64 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::UInt8x2 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::UInt8x4 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::UInt8x8 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::UInt8x16 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::UInt8x32 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::UInt8x64 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::UInt16x2 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::UInt16x4 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::UInt16x8 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::UInt16x16 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::UInt16x32 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::UInt16x64 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::UInt32x2 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::UInt32x4 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::UInt32x8 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::UInt32x16 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::UInt32x32 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::UInt32x64 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::UInt64x2 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::UInt64x4 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::UInt64x8 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::UInt64x16 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::UInt64x32 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::UInt64x64 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Int8x2 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Int8x4 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Int8x8 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Int8x16 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Int8x32 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Int8x64 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Int16x2 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Int16x4 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Int16x8 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Int16x16 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Int16x32 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Int16x64 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Int32x2 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Int32x4 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Int32x8 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Int32x16 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Int32x32 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Int32x64 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Int64x2 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Int64x4 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Int64x8 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Int64x16 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Int64x32 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Int64x64 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E3M4 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E3M4x2 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E3M4x4 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E3M4x8 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E3M4x16 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E3M4x32 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E3M4x64 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E4M3 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E4M3x2 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E4M3x4 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E4M3x8 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E4M3x16 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E4M3x32 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E4M3x64 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E4M3B11FNUZ (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E4M3B11FNUZx2 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E4M3B11FNUZx4 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E4M3B11FNUZx8 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E4M3B11FNUZx16 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E4M3B11FNUZx32 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E4M3B11FNUZx64 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E4M3FN (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E4M3FNx2 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E4M3FNx4 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E4M3FNx8 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E4M3FNx16 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E4M3FNx32 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E4M3FNx64 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E4M3FNUZ (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E4M3FNUZx2 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E4M3FNUZx4 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E4M3FNUZx8 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E4M3FNUZx16 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E4M3FNUZx32 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E4M3FNUZx64 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E5M2 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E5M2x2 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E5M2x4 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E5M2x8 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E5M2x16 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E5M2x32 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E5M2x64 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E5M2FNUZ (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E5M2FNUZx2 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E5M2FNUZx4 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E5M2FNUZx8 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E5M2FNUZx16 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E5M2FNUZx32 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E5M2FNUZx64 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E8M0FNU (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E8M0FNUx2 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E8M0FNUx4 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E8M0FNUx8 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E8M0FNUx16 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E8M0FNUx32 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float8E8M0FNUx64 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float6E2M3FN (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float6E2M3FNx2 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float6E2M3FNx4 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float6E2M3FNx8 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float6E2M3FNx16 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float6E2M3FNx32 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float6E2M3FNx64 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float6E3M2FN (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float6E3M2FNx2 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float6E3M2FNx4 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float6E3M2FNx8 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float6E3M2FNx16 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float6E3M2FNx32 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float6E3M2FNx64 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float4E2M1FN (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float4E2M1FNx2 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float4E2M1FNx4 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float4E2M1FNx8 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float4E2M1FNx16 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float4E2M1FNx32 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Float4E2M1FNx64 (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Boolean (ffi::Optional< PrimExpr > expr=std::nullopt)
 
PrimExpr tvm::script::ir_builder::tirx::Void (ffi::Optional< PrimExpr > expr=std::nullopt)
 

Macro Definition Documentation

◆ TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST

#define TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST (   FuncName,
  DType 
)
Value:
inline PrimExpr FuncName(ffi::Optional<PrimExpr> expr = std::nullopt) { \
PrimType dtype = DType; \
return expr.has_value() ? tvm::prim::cast(dtype, expr.value()) \
: tvm::Var("", dtype).as_or_throw<PrimExpr>(); \
}
Managed reference to VarNode.
Definition expr.h:714
TVM_TIRX_DLL PrimExpr cast(PrimType t, PrimExpr value, Location loc=UnknownLoc())
cast value to type.

◆ TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST_LANES

#define TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST_LANES (   FuncName,
  Code,
  Size 
)
Value:
TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST(FuncName##x2, (PrimType(DLDataType{Code, Size, 2}))) \
TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST(FuncName##x4, (PrimType(DLDataType{Code, Size, 4}))); \
TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST(FuncName##x8, (PrimType(DLDataType{Code, Size, 8}))); \
TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST(FuncName##x16, (PrimType(DLDataType{Code, Size, 16}))); \
TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST(FuncName##x32, (PrimType(DLDataType{Code, Size, 32}))); \
TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST(FuncName##x64, (PrimType(DLDataType{Code, Size, 64})));
#define TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST(FuncName, DType)
Definition ir.h:130

◆ TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST_LANES_FIXED_SIZE

#define TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST_LANES_FIXED_SIZE (   DType,
  Code,
  Bits 
)
Value:
TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST(DType, (PrimType(DLDataType{Code, Bits, 1}))); \
TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST(DType##x2, (PrimType(DLDataType{Code, Bits, 2}))); \
TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST(DType##x4, (PrimType(DLDataType{Code, Bits, 4}))); \
TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST(DType##x8, (PrimType(DLDataType{Code, Bits, 8}))); \
TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST(DType##x16, (PrimType(DLDataType{Code, Bits, 16}))); \
TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST(DType##x32, (PrimType(DLDataType{Code, Bits, 32}))); \
TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST(DType##x64, (PrimType(DLDataType{Code, Bits, 64})));

◆ TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST_SIZES

#define TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST_SIZES (   DType,
  Code 
)
Value:
TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST(DType##8, (PrimType(DLDataType{Code, 8, 1}))); \
TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST(DType##16, (PrimType(DLDataType{Code, 16, 1}))); \
TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST(DType##32, (PrimType(DLDataType{Code, 32, 1}))); \
TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST(DType##64, (PrimType(DLDataType{Code, 64, 1})));

◆ TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST_SIZES_LANES

#define TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST_SIZES_LANES (   DType,
  Code 
)
Value:
TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST_LANES(DType##16, Code, 16); \
TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST_LANES(DType##32, Code, 32); \
TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST_LANES(DType##64, Code, 64);
#define TVM_TIRX_IR_BUILDER_DEF_DTYPE_CAST_LANES(FuncName, Code, Size)
Definition ir.h:148