MCPcopy Create free account

hub / github.com/NVIDIA/cuda-tile / functions

Functions974 in github.com/NVIDIA/cuda-tile

Functionceildivi
Integer ceiling division operation.
python/cuda_tile/dialects/cuda_tile_ops.py:4616
Functioncheck_data_type_binary
Decorator to check if the specified tile has the expected data type.
python/cuda_tile/dialects/cuda_tile_ops.py:1619
Functioncheck_data_type_unary
Decorator to check if the specified tile has the expected data type.
python/cuda_tile/dialects/cuda_tile_ops.py:1637
Functioncheck_same_type
Decorator to check if lhs and rhs have the same tile type.
python/cuda_tile/dialects/cuda_tile_ops.py:1607
MethodclassifyParameter
tools/cuda-tile-tblgen/BytecodeTypeAnalysis.cpp:23
Methodclassof
lib/Dialect/CudaTile/IR/Attributes.cpp:580
Methodclassof
lib/Dialect/CudaTile/IR/Types.cpp:48
Methodclassof
Implementation of `classof` that supports all of the potential terminator operations.
include/cuda_tile/Dialect/CudaTile/IR/Ops.h:78
MethodcloneWith
lib/Dialect/CudaTile/IR/Types.cpp:207
Functioncmp
Performs element-wise comparison of two tiles.
python/cuda_tile/dialects/cuda_tile_ops.py:4506
Functioncos
Computes the cosine of the source tile element-wise.
python/cuda_tile/dialects/cuda_tile_ops.py:2377
Functioncosh
Computes the hyperbolic cosine of the source tile.
python/cuda_tile/dialects/cuda_tile_ops.py:2413
FunctioncovertSyntaxToSignature
tools/cuda-tile-tblgen/SpecGen.cpp:207
MethodcreateOperationGeneric
===----------------------------------------------------------------------===// Helper for Operation Creation and Result Handling ===------------------
lib/Bytecode/Reader/BytecodeReader.cpp:1138
Functioncreate_and_get_global
Helper function that inserts a new cuda_tile.global in the enclosing module and a cuda_tile.get_global at the current insertion point.
python/cuda_tile/dialects/cuda_tile_ops.py:4745
Methodcreate_entry
Creates a kernel entry in the CUDA Tile IR module. Args: kernel_name (str): The name of the kernel entry. ar
python/cuda_tile/dialects/cuda_tile_ops.py:5100
Methodcreate_tile_ir_module
Creates a CUDA Tile IR module.
python/cuda_tile/dialects/cuda_tile_ops.py:5086
Functioncuda_tile_op
This is a decorator that needs to be used in each cuda_tile OP to manage pre-generation things. Currently, it only generate source locati
python/cuda_tile/dialects/cuda_tile_ops.py:1208
Methoddanger
tools/cuda-tile-tblgen/Emitter.h:184
MethoddangerLine
tools/cuda-tile-tblgen/Emitter.h:188
Functiondecorator
(func)
python/cuda_tile/dialects/cuda_tile_ops.py:1622
Methoddim_map
(self)
python/cuda_tile/dialects/cuda_tile_ops.py:1136
Methoddim_map
(self)
python/cuda_tile/dialects/cuda_tile_ops.py:1175
MethoddispatchOpWriter
Dispatch to the correct op writer. Returns the number of results that were serialized.
lib/Bytecode/Writer/BytecodeWriter.cpp:1305
Methodelement_type
(self)
python/cuda_tile/dialects/cuda_tile_ops.py:890
Methodelement_type
(self)
python/cuda_tile/dialects/cuda_tile_ops.py:1084
MethodemitInclude
tools/cuda-tile-tblgen/Emitter.h:279
FunctionemitNamedConstraint
tools/cuda-tile-tblgen/SpecGen.cpp:199
FunctionemitSummary
tools/cuda-tile-tblgen/SpecGen.cpp:188
Functionexp
Raises e to the power of source.
python/cuda_tile/dialects/cuda_tile_ops.py:2470
Functionexp2
Raises 2 to the power of source.
python/cuda_tile/dialects/cuda_tile_ops.py:2450
Functionexti
( el_type, from_, *, signedness: Signedness = Signedness.SIGNED, loc=None, ip=None )
python/cuda_tile/dialects/cuda_tile_ops.py:3661
Functionextract
Extracts a slice from the source tile at the specified indices.
python/cuda_tile/dialects/cuda_tile_ops.py:2957
MethodextractCudaTileModuleOp
lib/Dialect/CudaTile/IR/CudaTile.cpp:62
MethodextractDefaultValue
tools/cuda-tile-tblgen/BytecodeGenUtilities.cpp:70
MethodextractVersionFromAttribute
tools/cuda-tile-tblgen/BytecodeGenUtilities.cpp:35
MethodextractVersionFromOperand
tools/cuda-tile-tblgen/BytecodeGenUtilities.cpp:86
MethodextractVersionFromOperation
tools/cuda-tile-tblgen/BytecodeGenUtilities.cpp:77
MethodextractVersionFromResult
tools/cuda-tile-tblgen/BytecodeGenUtilities.cpp:183
Methodflag
tools/cuda-tile-tblgen/CudaTileType.cpp:215
Methodfloat_tile
tools/cuda-tile-tblgen/CudaTileType.cpp:143
Functionfloor
Computes the floor of the source tile element-wise.
python/cuda_tile/dialects/cuda_tile_ops.py:2406
Functionfma
( lhs: Tile, rhs: Tile, acc: Tile, *, rounding_mode: RoundingMode = RoundingMode.NEAREST_E
python/cuda_tile/dialects/cuda_tile_ops.py:1861
Methodfold
lib/Dialect/CudaTile/IR/CudaTile.cpp:1484
Functionfor_loop
Constructs a for loop with the provided body. The body is a function taking as argument the iteration variables and building the operations w
python/cuda_tile/dialects/cuda_tile_ops.py:3937
MethodfromBuffer
include/cuda_tile/Dialect/CudaTile/Optimizer/CudaTileOptimizer.h:77
MethodfromFile
include/cuda_tile/Dialect/CudaTile/Optimizer/CudaTileOptimizer.h:83
MethodfromTableGen
tools/cuda-tile-tblgen/CudaTileAttr.cpp:56
MethodfromVersion
lib/Bytecode/Common/Version.cpp:25
Functionftof
( el_type, from_, *, rounding_mode: RoundingMode = RoundingMode.NEAREST_EVEN, loc=None,
python/cuda_tile/dialects/cuda_tile_ops.py:3578
Functionftoi
( el_type, from_, *, signedness: Signedness = Signedness.SIGNED, loc=None, ip=None )
python/cuda_tile/dialects/cuda_tile_ops.py:3620
MethodgenerateAttrTagEnum
tools/cuda-tile-tblgen/BytecodeAttrCodeGen.cpp:31
MethodgenerateAttrVersionCheck
tools/cuda-tile-tblgen/BytecodeAttrCodeGen.cpp:53
MethodgenerateDependentTypeRegistration
tools/cuda-tile-tblgen/BytecodeTypeCodeGen.cpp:800
MethodgenerateDeserializerDispatch
tools/cuda-tile-tblgen/BytecodeTypeCodeGen.cpp:825
MethodgenerateEnumAttrTypeTrait
tools/cuda-tile-tblgen/BytecodeAttrCodeGen.cpp:111
MethodgenerateEnumAttrVersionCheck
tools/cuda-tile-tblgen/BytecodeAttrCodeGen.cpp:151
MethodgenerateEnumTypeTrait
tools/cuda-tile-tblgen/BytecodeAttrCodeGen.cpp:92
MethodgenerateEnumValueVersionCheck
tools/cuda-tile-tblgen/BytecodeAttrCodeGen.cpp:183
MethodgenerateSerializerDispatch
tools/cuda-tile-tblgen/BytecodeTypeCodeGen.cpp:759
MethodgenerateSpec
tools/cuda-tile-tblgen/SpecGen.cpp:690
MethodgenerateTypeDeserializers
tools/cuda-tile-tblgen/BytecodeTypeCodeGen.cpp:747
MethodgenerateTypeSerializers
tools/cuda-tile-tblgen/BytecodeTypeCodeGen.cpp:522
MethodgenerateTypeTagEnum
tools/cuda-tile-tblgen/BytecodeTypeCodeGen.cpp:196
MethodgetAlias
Provide custom aliasing for debug info attributes.
lib/Dialect/CudaTile/IR/CudaTile.cpp:5629
MethodgetAllowTMA
lib/Dialect/CudaTile/IR/Attributes.cpp:333
MethodgetAsmBlockArgumentNames
lib/Dialect/CudaTile/IR/CudaTile.cpp:2618
MethodgetAsmResultNames
lib/Dialect/CudaTile/IR/CudaTile.cpp:1468
MethodgetAttributeForSmOrDefault
Helper function to retrieve an attribute from SM-specific or default entry. Returns the attribute if found, otherwise std::nullopt.
lib/Dialect/CudaTile/IR/Attributes.cpp:181
MethodgetBytecodeSize
lib/Bytecode/Reader/BytecodeReader.cpp:2416
MethodgetCanonicalizationPatterns
lib/Dialect/CudaTile/IR/CudaTile.cpp:3524
FunctiongetCheckedType
lib/CAPI/Dialect/CudaTileDialect.cpp:37
MethodgetConstantIndex
Look up a constant by attribute without adding it
lib/Bytecode/Writer/BytecodeWriter.cpp:484
MethodgetCudaTileSpecSubGroup
tools/cuda-tile-tblgen/CudaTileOp.cpp:421
MethodgetCurrentBytecodeVersion
lib/Bytecode/Common/CommandLineOptions.cpp:84
MethodgetDILoc
Returns a CudaTile location for a given location attribute.
lib/Dialect/CudaTile/IR/CudaTile.cpp:1234
MethodgetDebugInfo
lib/Bytecode/Reader/BytecodeReader.cpp:844
MethodgetDebugInfoIndex
This method gets or creates an index for a debug info attribute.
lib/Bytecode/Writer/BytecodeWriter.cpp:728
MethodgetDebugReserved
lib/Bytecode/Writer/BytecodeWriter.cpp:764
MethodgetDescription
test/lib/TestPasses.cpp:40
MethodgetErrorUnsupportedHints
lib/Bytecode/Common/CommandLineOptions.cpp:107
MethodgetI1SameShape
lib/Dialect/CudaTile/IR/Types.cpp:214
MethodgetLatency
lib/Dialect/CudaTile/IR/Attributes.cpp:343
MethodgetMaxSignedValueForBitwidth
lib/Dialect/CudaTile/IR/CudaTile.cpp:41
MethodgetMaxUnsignedValueForBitwidth
lib/Dialect/CudaTile/IR/CudaTile.cpp:55
MethodgetMinSignedValueForBitwidth
lib/Dialect/CudaTile/IR/CudaTile.cpp:48
MethodgetNumCTAInCGA
lib/Dialect/CudaTile/IR/Attributes.cpp:305
MethodgetNumWorkerWarpsPerCTA
lib/Dialect/CudaTile/IR/Attributes.cpp:318
MethodgetOccupancy
lib/Dialect/CudaTile/IR/Attributes.cpp:356
MethodgetOpcodeForOperation
lib/Bytecode/Writer/BytecodeWriter.cpp:988
MethodgetOperationName
include/cuda_tile/Dialect/CudaTile/IR/Ops.h:92
MethodgetSubprogram
Returns a subprogram attribute for a given local scope attribute.
lib/Dialect/CudaTile/IR/CudaTile.cpp:1224
MethodgetVersionOrderedBitAssignments
tools/cuda-tile-tblgen/BytecodeGenUtilities.cpp:129
Functionget_index_space_shape
( view: TileView, result_type=Int64, loc=None, ip=None )
python/cuda_tile/dialects/cuda_tile_ops.py:4780
Functionget_num_tile_blocks
Get number of tile blocks.
python/cuda_tile/dialects/cuda_tile_ops.py:3016
Functionget_tensor_shape
( view: TensorView, result_type=Int64, loc=None, ip=None )
python/cuda_tile/dialects/cuda_tile_ops.py:4794
Functionget_tile_block_id
Get the ID of the current tile block.
python/cuda_tile/dialects/cuda_tile_ops.py:3010
MethodhandleTerminator
lib/Dialect/CudaTile/IR/CudaTile.cpp:5561
Functionif_generate
( condition: Tile, then_body: Callable, else_body: Optional[Callable] | None = None, input_arg
python/cuda_tile/dialects/cuda_tile_ops.py:3733
Methodindex_type
Returns the MLIR index type for this tensor view.
python/cuda_tile/dialects/cuda_tile_ops.py:1096
← previousnext →601–700 of 974, ranked by callers