Searched refs:ptx_type (Results 1 – 2 of 2) sorted by relevance
| /external/llvm-project/clang/test/CodeGen/ |
| D | builtins-nvptx-mma.py | 19 self.ptx_type = ptx_elt_type;
22 return "%s:%s:%s" % (self.geom, self.frag, self.ptx_type)
46 return [MMAFrag(geom, frag, ptx_type) for (geom, frag, ptx_type)
78 def is_type_supported(ptx_type): argument
79 if ptx_type in ["s8", "u8", "s32"]:
81 if ptx_type in ["s4", "u4", "b1"]:
86 if not (is_type_supported(op.a.ptx_type)
90 if op.a.ptx_type in ["s4", "u4", "b1"]:
91 if op.a.ptx_type == "b1" and satf:
97 if not (is_type_supported(frag.ptx_type)
[all …]
|
| /external/llvm-project/llvm/test/CodeGen/NVPTX/ |
| D | wmma.py | 56 def __init__(self, ptx_type): argument
57 self.ptx_type = ptx_type
67 }[ptx_type];
72 }.get(ptx_type, "%r[0-9]+")
75 return "%s/%s" % (self.ptx_type, self.llvm_type)
153 return [MMAFrag(geom, frag, ptx_type) for (geom, frag, ptx_type)
190 def is_type_supported(ptx_type): argument
191 if ptx_type in ["s8", "u8", "s32"]:
193 if ptx_type in ["s4", "u4", "b1"]:
199 if not (is_type_supported(op.a.mma_type.ptx_type)
[all …]
|