Python uniqueByの例

コード例 #1

ファイル: store.py プロジェクト: JDJFisher/op-cg

 def loops(self) -> List[OP.Loop]:
     loops = flattern(program.loops for program in self.programs)
     return uniqueBy(loops, lambda l: l.kernel)

コード例 #2

ファイル: store.py プロジェクト: JDJFisher/op-cg

 def consts(self) -> List[OP.Const]:
     consts = flattern(program.consts for program in self.programs)
     return uniqueBy(consts, lambda c: c.ptr)

コード例 #3

ファイル: op.py プロジェクト: JDJFisher/op-cg

 def indirectIdxs(self) -> List[Arg]:
   return uniqueBy(self.indirects, lambda a: (a.map, a.idx))

コード例 #4

ファイル: op.py プロジェクト: JDJFisher/op-cg

 def indirectMaps(self) -> List[Arg]:
   return uniqueBy(self.indirects, lambda a: a.map)

コード例 #5

ファイル: op.py プロジェクト: JDJFisher/op-cg

 def indirectVars(self) -> List[Arg]:
   return uniqueBy(self.indirects, lambda a: a.var)

コード例 #6

ファイル: op.py プロジェクト: JDJFisher/op-cg

 def uniqueVars(self) -> List[Arg]:
   return uniqueBy(self.args, lambda a: a.var)