Lines Matching refs:PartialSchedule

316 static bool containsOnlyMatrMultAcc(isl::map PartialSchedule,  in containsOnlyMatrMultAcc()  argument
318 auto InputDimId = PartialSchedule.get_tuple_id(isl::dim::in); in containsOnlyMatrMultAcc()
320 unsigned OutDimNum = unsignedFromIslSize(PartialSchedule.range_tuple_dim()); in containsOnlyMatrMultAcc()
325 permuteDimensions(PartialSchedule, isl::dim::out, MMI.i, OutDimNum - 1); in containsOnlyMatrMultAcc()
327 permuteDimensions(PartialSchedule, isl::dim::out, MMI.j, OutDimNum - 1); in containsOnlyMatrMultAcc()
329 permuteDimensions(PartialSchedule, isl::dim::out, MMI.k, OutDimNum - 1); in containsOnlyMatrMultAcc()
401 static bool containsMatrMult(isl::map PartialSchedule, const Dependences *D, in containsMatrMult() argument
403 auto InputDimsId = PartialSchedule.get_tuple_id(isl::dim::in); in containsMatrMult()
422 if (!containsOnlyMatMulDep(PartialSchedule, D, MMI.k)) in containsMatrMult()
425 if (!MMI.WriteToC || !containsOnlyMatrMultAcc(PartialSchedule, MMI)) in containsMatrMult()
446 auto PartialSchedule = in permuteBandNodeDimensions() local
448 auto PartialScheduleFirstDim = PartialSchedule.at(FirstDim); in permuteBandNodeDimensions()
449 auto PartialScheduleSecondDim = PartialSchedule.at(SecondDim); in permuteBandNodeDimensions()
450 PartialSchedule = in permuteBandNodeDimensions()
451 PartialSchedule.set_union_pw_aff(SecondDim, PartialScheduleFirstDim); in permuteBandNodeDimensions()
452 PartialSchedule = in permuteBandNodeDimensions()
453 PartialSchedule.set_union_pw_aff(FirstDim, PartialScheduleSecondDim); in permuteBandNodeDimensions()
455 return Node.insert_partial_schedule(PartialSchedule); in permuteBandNodeDimensions()
1008 auto PartialSchedule = isl::manage( in isMatrMultPattern() local
1016 isl_union_map_n_map(PartialSchedule.get()) != 1) in isMatrMultPattern()
1018 auto NewPartialSchedule = isl::map::from_union_map(PartialSchedule); in isMatrMultPattern()