def _register()

in python/featgraph/module/spmm.py [0:0]

• 9 lines of code
• 2 McCabe index (conditional complexity)

    def _register(self):
        self._target = 'llvm'
        self._ctx = tvm.cpu(0)
        if self._num_col_partitions > 1:
            self._compute_func = vanilla_spmm_dds_x86
            self._schedule_func = schedule_vanilla_spmm_dds_x86
        else:
            self._compute_func = vanilla_spmm_csr_x86
            self._schedule_func = schedule_vanilla_spmm_csr_x86