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

Method generate

tools/gen_header_for_bin_reduce.py:97–110  ·  view source on GitHub ↗
(self, fout)

Source from the content-addressed store, hash-verified

95 )
96
97 def generate(self, fout):
98 self._fout = fout
99 self._write_def("MGB_BINREDUCE_VERSION", "20220507")
100 if self._has_netinfo:
101 self._write_dtype()
102
103 if len(self._elemwise_modes) > 0:
104 self._write_elemwise_modes()
105
106 if self._has_netinfo:
107 self._write_oprs()
108 self._write_hash()
109 self._write_midout()
110 del self._fout
111
112 def strip_opr_name_with_version(self, name):
113 pos = len(name)

Callers 1

mainFunction · 0.95

Calls 6

_write_defMethod · 0.95
_write_dtypeMethod · 0.95
_write_elemwise_modesMethod · 0.95
_write_oprsMethod · 0.95
_write_hashMethod · 0.95
_write_midoutMethod · 0.95

Tested by

no test coverage detected