def gpu_schedule_Tile(outs): """ gpu schedule for tile. Args: outs (tvm.tensor.Tensor): outputs of compute. Returns: sch (schedule.Schedule): The created schedule. """ device = 'cuda' ctx = _akg.tvm.context(device, 0) if not ctx.exist: raise SystemError("Skip because %s is not enabled" % device) with _akg.tvm.target.create(device): s = schedule_elemwise(outs) return s
def gpu_schedule_Cast(outs): """ gpu schedule for cast. Args: outs (tvm.tensor.Tensor): outputs of compute. Returns: sch (schedule.Schedule): The created schedule. """ device = 'cuda' ctx = _akg.tvm.context(device, 0) if not ctx.exist: logging.info("Skip because %s is not enabled", device) return None with _akg.tvm.target.create(device): sch = schedule_elemwise(outs) return sch