Method init

cutlass.py/tile_scheduler.py:402–406

(self, M, N)

        valid: bool

    def init(self, M, N):
        self.linear_idx = (
            dev_coord.blockIdx.x + dev_coord.blockIdx.y * dev_coord.gridDim.x
        )
        self.get_blocks_m_n(M, N)

    def get_current_work_info(self):
        m_idx, n_idx = self.get_current_m_n_idx()