Lines Matching defs:predicate
73 def debug_print(fmt, *args, predicate=None, threadNumber=-1, forcePrint=False):
96 predicate = arith.cmpi(arith.CmpIPredicate.eq, tidx, c(threadNumber))
98 if_op = scf.IfOp(predicate)
338 nvgpu.mbarrier_init(mbarTMA, c(1), c(i), predicate=wgPrimaryThread)
339 nvgpu.mbarrier_init(mbarDONE, c(1), c(i), predicate=wgPrimaryThread)
343 nvgpu.tma_prefetch_descriptor(a_tma_desc_op, predicate=wgPrimaryThread)
344 nvgpu.tma_prefetch_descriptor(b_tma_desc_op, predicate=wgPrimaryThread)
369 predicate=producerPrimaryThread,
379 predicate=producerPrimaryThread,
427 predicate=producerPrimaryThread,
436 predicate=producerPrimaryThread,
444 predicate=producerPrimaryThread,
453 predicate=producerPrimaryThread,
461 predicate=producerPrimaryThread,
464 mbarTMA, c(txcount), stage, predicate=producerPrimaryThread
470 predicate=producerPrimaryThread,
500 predicate=consumerPrimaryThread,
510 predicate=consumerPrimaryThread,
539 predicate=consumerPrimaryThread,
569 predicate=consumerPrimaryThread,
576 predicate=consumerPrimaryThread,
600 debug_print("[cons] | Storing", predicate=consumerPrimaryThread)
819 nvgpu.mbarrier_init(mbarTMA, c(1), c(i), predicate=primaryThread)
823 nvgpu.tma_prefetch_descriptor(a_tma_desc_op, predicate=primaryThread)
824 nvgpu.tma_prefetch_descriptor(b_tma_desc_op, predicate=primaryThread)
879 predicate=primaryThread,
887 predicate=primaryThread,
895 predicate=primaryThread,
903 predicate=primaryThread,
908 "[Prologue] mbarTMA[{}] arrive", iv, predicate=primaryThread
911 mbarTMA, c(txcount), iv, predicate=primaryThread
916 predicate=primaryThread,
934 predicate=primaryThread,
943 predicate=primaryThread,
973 predicate=primaryThread,
1004 predicate=p,
1007 mbarTMA, c(txcount), nextSlot, predicate=p
1012 predicate=p,
1059 predicate=p,
1067 predicate=p,
1075 predicate=p,
1084 predicate=p,
1105 debug_print("Storing", predicate=primaryThread)