1 // 2 // Copyright 2019 Red Hat, Inc. 3 // 4 // Permission is hereby granted, free of charge, to any person obtaining a 5 // copy of this software and associated documentation files (the "Software"), 6 // to deal in the Software without restriction, including without limitation 7 // the rights to use, copy, modify, merge, publish, distribute, sublicense, 8 // and/or sell copies of the Software, and to permit persons to whom the 9 // Software is furnished to do so, subject to the following conditions: 10 // 11 // The above copyright notice and this permission notice shall be included in 12 // all copies or substantial portions of the Software. 13 // 14 // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR 15 // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, 16 // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL 17 // THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR 18 // OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, 19 // ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR 20 // OTHER DEALINGS IN THE SOFTWARE. 21 // 22 23 #ifndef CLOVER_CORE_COMPILER_HPP 24 #define CLOVER_CORE_COMPILER_HPP 25 26 #include "core/device.hpp" 27 #include "core/binary.hpp" 28 #include "llvm/invocation.hpp" 29 #include "nir/invocation.hpp" 30 #include "spirv/invocation.hpp" 31 32 namespace clover { 33 namespace compiler { 34 static inline binary compile_program(const program & prog,const header_map & headers,const device & dev,const std::string & opts,std::string & log)35 compile_program(const program &prog, const header_map &headers, 36 const device &dev, const std::string &opts, 37 std::string &log) { 38 switch (dev.ir_format()) { 39 #ifdef HAVE_CLOVER_SPIRV 40 case PIPE_SHADER_IR_NIR_SERIALIZED: 41 switch (prog.il_type()) { 42 case program::il_type::source: 43 return llvm::compile_to_spirv(prog.source(), headers, dev, opts, log); 44 case program::il_type::spirv: 45 return spirv::compile_program(prog.source(), dev, log); 46 default: 47 unreachable("device with unsupported IL"); 48 throw error(CL_INVALID_VALUE); 49 } 50 #endif 51 case PIPE_SHADER_IR_NATIVE: 52 if (prog.il_type() == program::il_type::source) 53 return llvm::compile_program(prog.source(), headers, dev, opts, log); 54 else 55 throw error(CL_INVALID_VALUE); 56 default: 57 unreachable("device with unsupported IR"); 58 throw error(CL_INVALID_VALUE); 59 } 60 } 61 62 static inline binary link_program(const std::vector<binary> & bs,const device & dev,const std::string & opts,std::string & log)63 link_program(const std::vector<binary> &bs, const device &dev, 64 const std::string &opts, std::string &log) { 65 switch (dev.ir_format()) { 66 #ifdef HAVE_CLOVER_SPIRV 67 case PIPE_SHADER_IR_NIR_SERIALIZED: { 68 const bool create_library = 69 opts.find("-create-library") != std::string::npos; 70 auto spirv_linked_module = spirv::link_program(bs, dev, opts, log); 71 if (create_library) 72 return spirv_linked_module; 73 return nir::spirv_to_nir(spirv_linked_module, 74 dev, log); 75 } 76 #endif 77 case PIPE_SHADER_IR_NATIVE: 78 return llvm::link_program(bs, dev, opts, log); 79 default: 80 unreachable("device with unsupported IR"); 81 throw error(CL_INVALID_VALUE); 82 } 83 } 84 } 85 } 86 87 #endif 88