def gpu_schedule_Equal(outs): """ gpu schedule for Equal. 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): sch = schedule_elemwise(outs) return sch
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