FreeTensor
Loading...
Searching...
No Matches
Public Member Functions | Public Attributes | List of all members
freetensor::CutlassMicroKernelProperty Struct Reference

#include <cutlass_micro_kernel_property.h>

Inheritance diagram for freetensor::CutlassMicroKernelProperty:
Inheritance graph
[legend]
Collaboration diagram for freetensor::CutlassMicroKernelProperty:
Collaboration graph
[legend]

Public Member Functions

template<typename TwarpIdBatch , typename TwarpIdM , typename TwarpIdN , typename TlaneId >
 CutlassMicroKernelProperty (int nWarpBatch, int nWarpM, int nWarpN, TwarpIdBatch &&warpIdBatch, TwarpIdM &&warpIdM, TwarpIdN &&warpIdN, TlaneId &&laneId)
 
void compHash () override
 
- Public Member Functions inherited from freetensor::ASTPart
virtual ~ASTPart ()
 
 ASTPart (ASTPart &&other)
 
 ASTPart (const ASTPart &other)
 
ASTPartoperator= (ASTPart &&)
 
ASTPartoperator= (const ASTPart &)
 
bool trySetParent (const Ref< ASTPart > &parent)
 
void resetParent ()
 
Ref< ASTPartparent () const
 
bool isSubTree () const
 
int depth () const
 
virtual void modifiedHook ()
 
size_t hash ()
 
virtual bool isAST () const
 
- Public Member Functions inherited from freetensor::EnableSelf< ASTPart >
Ref< ASTPartself () const
 

Public Attributes

int nWarpBatch_
 
int nWarpM_
 
int nWarpN_
 
Expr warpIdBatch_
 
Expr warpIdM_
 
Expr warpIdN_
 
Expr laneId_
 

Additional Inherited Members

- Public Types inherited from freetensor::EnableSelf< ASTPart >
typedef ASTPart Self
 
- Protected Member Functions inherited from freetensor::ASTPart
void lock ()
 
void unlock ()
 
virtual void compHash ()=0
 
void resetHash ()
 
- Protected Attributes inherited from freetensor::ASTPart
size_t hash_ = ~0ull
 
std::atomic_flag lock_ = ATOMIC_FLAG_INIT
 

Constructor & Destructor Documentation

◆ CutlassMicroKernelProperty()

template<typename TwarpIdBatch , typename TwarpIdM , typename TwarpIdN , typename TlaneId >
freetensor::CutlassMicroKernelProperty::CutlassMicroKernelProperty ( int  nWarpBatch,
int  nWarpM,
int  nWarpN,
TwarpIdBatch &&  warpIdBatch,
TwarpIdM &&  warpIdM,
TwarpIdN &&  warpIdN,
TlaneId &&  laneId 
)
inline

Member Function Documentation

◆ compHash()

void freetensor::CutlassMicroKernelProperty::compHash ( )
overridevirtual

Implements freetensor::ASTPart.

Member Data Documentation

◆ laneId_

Expr freetensor::CutlassMicroKernelProperty::laneId_

◆ nWarpBatch_

int freetensor::CutlassMicroKernelProperty::nWarpBatch_

◆ nWarpM_

int freetensor::CutlassMicroKernelProperty::nWarpM_

◆ nWarpN_

int freetensor::CutlassMicroKernelProperty::nWarpN_

◆ warpIdBatch_

Expr freetensor::CutlassMicroKernelProperty::warpIdBatch_

◆ warpIdM_

Expr freetensor::CutlassMicroKernelProperty::warpIdM_

◆ warpIdN_

Expr freetensor::CutlassMicroKernelProperty::warpIdN_

The documentation for this struct was generated from the following files: