MCPcopy Create free account
hub / github.com/MegEngine/MegEngine / MathInstruction

Class MathInstruction

dnn/scripts/cutlass_generator/library.py:637–656  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

635
636#
637class MathInstruction:
638 def __init__(
639 self,
640 instruction_shape,
641 element_a,
642 element_b,
643 element_accumulator,
644 opcode_class,
645 math_operation=MathOperation.multiply_add,
646 element_rin=DataType.s32,
647 element_rout=DataType.s32,
648 ):
649 self.instruction_shape = instruction_shape
650 self.element_a = element_a
651 self.element_b = element_b
652 self.element_accumulator = element_accumulator
653 self.opcode_class = opcode_class
654 self.math_operation = math_operation
655 self.element_rin = element_rin
656 self.element_rout = element_rout
657
658
659#

Callers 12

GenerateConv2d_SimtFunction · 0.85
GenerateDeconv_SimtFunction · 0.85
GenerateGemm_SimtFunction · 0.85
GenerateDwconv2d_SimtFunction · 0.85
GenerateGemv_SimtFunction · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected