Searched defs:LLVMCodeGenBuilder (Results 1 – 1 of 1) sorted by relevance
82 struct TORCH_API LLVMCodeGenBuilder { struct85 LLVMCodeGenBuilder(StmtPtr stmt, std::vector<BufferArg> args) in LLVMCodeGenBuilder() argument88 LLVMCodeGenBuilder& device(at::Device device) { in device() argument93 LLVMCodeGenBuilder& kernelFuncName(std::string name) { in kernelFuncName() argument98 LLVMCodeGenBuilder& dtype(Dtype d) { in dtype() argument103 LLVMCodeGenBuilder& triple(std::string triple) { in triple() argument108 LLVMCodeGenBuilder& cpu(std::string cpu) { in cpu() argument113 LLVMCodeGenBuilder& attrs(std::string attrs) { in attrs() argument118 std::unique_ptr<LLVMCodeGen> build() { in build()125 std::vector<BufferArg> args_;[all …]