autogen.h

Go to the documentation of this file.

#pragma once
 
#include <mim/axm.h>
#include <mim/plugin.h>
 
/// @namespace mim::plug::btensor @ref btensor
namespace mim {
namespace plug::btensor {
 
static constexpr plugin_t Plugin_Id = 0xd518f5104c00000;
 
/// @name `btensor.Ring`
///@{
enum class Ring : flags_t {
};
 
/// The fields of `btensor.Ring`, by index.
struct Ring_field { enum : nat_t { T = 0, _0 = 1, add = 2, mul = 3, }; };
 
///@}
 
/// @name `btensor.id`
///@{
enum class id : flags_t {
};
 
///@}
 
/// @name `btensor.map_reduce`
///@{
enum class map_reduce : flags_t {
};
 
const Def* normalize_map_reduce(const Def*, const Def*, const Def*);
///@}
 
/// @name `btensor.map_reduce_post`
///@{
enum class map_reduce_post : flags_t {
};
 
///@}
 
/// @name `btensor.Sched`
///@{
enum class Sched : flags_t {
};
 
/// The fields of `btensor.Sched`, by index.
struct Sched_field { enum : nat_t { vdim = 0, unroll = 1, }; };
 
///@}
 
/// @name `btensor.SchedT`
///@{
enum class SchedT : flags_t {
};
 
///@}
 
/// @name `btensor.mk_sched`
///@{
enum class mk_sched : flags_t {
};
 
///@}
 
/// @name `btensor.proj_map`
///@{
enum class proj_map : flags_t {
};
 
///@}
 
/// @name `btensor.broadcast`
///@{
enum class broadcast : flags_t {
};
 
///@}
 
/// @name `btensor.concat`
///@{
enum class concat : flags_t {
};
 
///@}
 
/// @name `btensor.pad`
///@{
enum class pad : flags_t {
};
 
///@}
 
/// @name `btensor.gather`
///@{
enum class gather : flags_t {
};
 
///@}
 
/// @name `btensor.scatter`
///@{
enum class scatter : flags_t {
};
 
///@}
 
/// @name `btensor.NestT`
///@{
enum class NestT : flags_t {
};
 
///@}
 
/// @name `btensor.mr_nest`
///@{
enum class mr_nest : flags_t {
};
 
///@}
 
/// @name `btensor.map_reduce_post_impl`
///@{
enum class map_reduce_post_impl : flags_t {
};
 
///@}
 
/// @name `btensor.lower_map_reduce`
///@{
enum class lower_map_reduce : flags_t {
};
 
///@}
 
void register_normalizers(Normalizers& normalizers);
 
#define MIM_btensor_NORMALIZER_IMPL \
    void register_normalizers(Normalizers& normalizers) {\
        normalizers[flags_t(Annex::Base<map_reduce>)] = &normalize_map_reduce; \
    }
} // namespace plug::btensor
 
#ifndef DOXYGEN // don't include in Doxygen documentation
 
template<> constexpr flags_t Annex::Base<plug::btensor::Ring> = 0xd518f5104c00000;
template<> constexpr size_t Annex::Num<plug::btensor::Ring> = 0;
template<> constexpr flags_t Annex::Base<plug::btensor::id> = 0xd518f5104c00100;
template<> constexpr size_t Annex::Num<plug::btensor::id> = 0;
template<> constexpr flags_t Annex::Base<plug::btensor::map_reduce> = 0xd518f5104c00200;
template<> constexpr size_t Annex::Num<plug::btensor::map_reduce> = 0;
template<> constexpr flags_t Annex::Base<plug::btensor::map_reduce_post> = 0xd518f5104c00300;
template<> constexpr size_t Annex::Num<plug::btensor::map_reduce_post> = 0;
template<> constexpr flags_t Annex::Base<plug::btensor::Sched> = 0xd518f5104c00400;
template<> constexpr size_t Annex::Num<plug::btensor::Sched> = 0;
template<> constexpr flags_t Annex::Base<plug::btensor::SchedT> = 0xd518f5104c00500;
template<> constexpr size_t Annex::Num<plug::btensor::SchedT> = 0;
template<> constexpr flags_t Annex::Base<plug::btensor::mk_sched> = 0xd518f5104c00600;
template<> constexpr size_t Annex::Num<plug::btensor::mk_sched> = 0;
template<> constexpr flags_t Annex::Base<plug::btensor::proj_map> = 0xd518f5104c00700;
template<> constexpr size_t Annex::Num<plug::btensor::proj_map> = 0;
template<> constexpr flags_t Annex::Base<plug::btensor::broadcast> = 0xd518f5104c00800;
template<> constexpr size_t Annex::Num<plug::btensor::broadcast> = 0;
template<> constexpr flags_t Annex::Base<plug::btensor::concat> = 0xd518f5104c00900;
template<> constexpr size_t Annex::Num<plug::btensor::concat> = 0;
template<> constexpr flags_t Annex::Base<plug::btensor::pad> = 0xd518f5104c00a00;
template<> constexpr size_t Annex::Num<plug::btensor::pad> = 0;
template<> constexpr flags_t Annex::Base<plug::btensor::gather> = 0xd518f5104c00b00;
template<> constexpr size_t Annex::Num<plug::btensor::gather> = 0;
template<> constexpr flags_t Annex::Base<plug::btensor::scatter> = 0xd518f5104c00c00;
template<> constexpr size_t Annex::Num<plug::btensor::scatter> = 0;
template<> constexpr flags_t Annex::Base<plug::btensor::NestT> = 0xd518f5104c00d00;
template<> constexpr size_t Annex::Num<plug::btensor::NestT> = 0;
template<> constexpr flags_t Annex::Base<plug::btensor::mr_nest> = 0xd518f5104c00e00;
template<> constexpr size_t Annex::Num<plug::btensor::mr_nest> = 0;
template<> constexpr flags_t Annex::Base<plug::btensor::map_reduce_post_impl> = 0xd518f5104c00f00;
template<> constexpr size_t Annex::Num<plug::btensor::map_reduce_post_impl> = 0;
template<> constexpr flags_t Annex::Base<plug::btensor::lower_map_reduce> = 0xd518f5104c01000;
template<> constexpr size_t Annex::Num<plug::btensor::lower_map_reduce> = 0;
 
template<> struct Axm::IsANode<plug::btensor::Ring> { using type = Axm; };
template<> struct Axm::IsANode<plug::btensor::id> { using type = Axm; };
template<> struct Axm::IsANode<plug::btensor::Sched> { using type = Axm; };
template<> struct Axm::IsANode<plug::btensor::SchedT> { using type = Axm; };
template<> struct Axm::IsANode<plug::btensor::mk_sched> { using type = Axm; };
template<> struct Axm::IsANode<plug::btensor::proj_map> { using type = Axm; };
template<> struct Axm::IsANode<plug::btensor::NestT> { using type = Axm; };
template<> struct Axm::IsANode<plug::btensor::mr_nest> { using type = Axm; };
template<> struct Axm::IsANode<plug::btensor::map_reduce_post_impl> { using type = Axm; };
template<> struct Axm::IsANode<plug::btensor::lower_map_reduce> { using type = Axm; };
 
#endif
} // namespace mim
 
#ifndef DOXYGEN // don't include in Doxygen documentation
 
 
#endif