OmniSciDB  72c90bc290
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Friends Macros Groups Pages
ResultSetReductionCodegen.h
Go to the documentation of this file.
1 /*
2  * Copyright 2022 HEAVY.AI, Inc.
3  *
4  * Licensed under the Apache License, Version 2.0 (the "License");
5  * you may not use this file except in compliance with the License.
6  * You may obtain a copy of the License at
7  *
8  * http://www.apache.org/licenses/LICENSE-2.0
9  *
10  * Unless required by applicable law or agreed to in writing, software
11  * distributed under the License is distributed on an "AS IS" BASIS,
12  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13  * See the License for the specific language governing permissions and
14  * limitations under the License.
15  */
16 
17 #pragma once
18 
19 #include "ResultSetReductionJIT.h"
20 #include "ResultSetReductionOps.h"
21 
22 #include <llvm/IR/LLVMContext.h>
23 #include <llvm/IR/Type.h>
24 
25 // Convert an IR type to the corresponding LLVM one.
26 llvm::Type* llvm_type(const Type type, llvm::LLVMContext& ctx);
27 
28 // Translate a function to a LLVM function provided as llvm_function (initially empty).
29 // The mapping to LLVM for the reduction functions is also provided as input f.
30 void translate_function(const Function* function,
31  llvm::Function* llvm_function,
32  const ReductionCode& reduction_code,
33  const std::unordered_map<const Function*, llvm::Function*>& f);
void translate_function(const Function *function, llvm::Function *llvm_function, const ReductionCode &reduction_code, const std::unordered_map< const Function *, llvm::Function * > &f)
llvm::Type * llvm_type(const Type type, llvm::LLVMContext &ctx)
torch::Tensor f(torch::Tensor x, torch::Tensor W_target, torch::Tensor b_target)