Home
last modified time | relevance | path

Searched refs:llvm_module (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_compiler.cc506 const llvm::Module& llvm_module) { in GetIRModuleHooks() argument
510 user_hook(llvm_module); in GetIRModuleHooks()
512 llvm_ir::DumpIrIfEnabled(*hlo_module_ptr, llvm_module, optimized); in GetIRModuleHooks()
514 return {[hook](const llvm::Module& llvm_module) { in GetIRModuleHooks()
515 return hook(/*optimized=*/false, llvm_module); in GetIRModuleHooks()
517 [hook](const llvm::Module& llvm_module) { in GetIRModuleHooks()
518 return hook(/*optimized=*/true, llvm_module); in GetIRModuleHooks()
522 Status VerifyLlvmModule(const llvm::Module& llvm_module) { in VerifyLlvmModule() argument
529 TF_RET_CHECK(!llvm::verifyModule(llvm_module, &err_stream)) in VerifyLlvmModule()
678 auto llvm_module = in RunBackend() local
[all …]
Dmlir_emitter.cc95 llvm::Module *llvm_module = b->GetInsertBlock()->getParent()->getParent(); in EmitMlirFuncAndCall() local
122 mlir_llvm_module->setDataLayout(llvm_module->getDataLayout()); in EmitMlirFuncAndCall()
124 *llvm_module, std::move(mlir_llvm_module), llvm::Linker::None, in EmitMlirFuncAndCall()
132 llvm::Function *func = llvm_module->getFunction(func_name); in EmitMlirFuncAndCall()
Dir_function.cc30 llvm::Module* llvm_module, const int64 num_dynamic_loop_bounds) { in GetComputeFunctionParams() argument
31 llvm::Type* i8_ptr_type = llvm::Type::getInt8PtrTy(llvm_module->getContext()); in GetComputeFunctionParams()
34 llvm::Type::getInt64PtrTy(llvm_module->getContext()); in GetComputeFunctionParams()
47 llvm::Module* llvm_module, llvm::IRBuilder<>* b, in IrFunction() argument
50 llvm_module_(llvm_module), in IrFunction()
Dir_function.h56 const HloModuleConfig& module_config, llvm::Module* llvm_module,
Dir_emitter.h85 const BufferAssignment& assignment, llvm::Module* llvm_module,
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_compiler.cc555 std::unique_ptr<llvm::Module>* llvm_module, in CompileModuleToLlvmIrImpl() argument
559 *llvm_module = absl::make_unique<llvm::Module>("", *llvm_context); in CompileModuleToLlvmIrImpl()
561 (*llvm_module)->setTargetTriple(target_triple); in CompileModuleToLlvmIrImpl()
562 (*llvm_module)->setDataLayout(data_layout); in CompileModuleToLlvmIrImpl()
599 llvm_module->get()); in CompileModuleToLlvmIrImpl()
677 std::unique_ptr<llvm::Module> llvm_module, in CompileToTargetBinary() argument
685 llvm::Module* llvm_module, bool relocatable, in CompileToTargetBinary()
691 llvm_module->getContext().setDiagnosticHandlerCallBack( in CompileToTargetBinary()
698 TF_RET_CHECK(!llvm::verifyModule(*llvm_module, &err_stream)) in CompileToTargetBinary()
710 CompileTargetBinary(module_config, llvm_module, gpu_version, in CompileToTargetBinary()
[all …]
Dir_emitter_context.h45 mlir::MLIRContext* mlir_context, llvm::Module* llvm_module) in IrEmitterContext() argument
53 llvm_module_(llvm_module) {} in IrEmitterContext()
70 llvm::Module* llvm_module() { return llvm_module_; } in llvm_module() function
Dgpu_compiler.h90 llvm::Module* llvm_module, GpuVersion gpu_version,
106 std::unique_ptr<llvm::Module> llvm_module,
172 std::unique_ptr<llvm::Module> llvm_module,
Dhlo_to_ir_bindings.h39 llvm::IRBuilder<>* b, llvm::Module* llvm_module, in HloToIrBindings() argument
44 module_(llvm_module) {} in HloToIrBindings()
Damdgpu_compiler.cc115 llvm::Module* llvm_module, in CompileTargetBinary() argument
134 hsaco, amdgpu::CompileToHsaco(llvm_module, gpu_version, module_config, in CompileTargetBinary()
Damdgpu_compiler.h43 const HloModuleConfig& module_config, llvm::Module* llvm_module,
Dir_emitter_nested.cc77 root_shape, ir_emitter_context_->llvm_module()->getDataLayout()); in CodegenNestedComputation()
89 ir_emitter_context_->llvm_module()); // The parent LLVM module. in CodegenNestedComputation()
Dir_emitter_unnested.cc159 llvm::Module* llvm_module) { in UpdateLaunchDimensions() argument
167 llvm_module->getOrInsertNamedMetadata("nvvm.annotations"); in UpdateLaunchDimensions()
169 llvm_module->getFunction(kernel_thunk->kernel_name().c_str()); in UpdateLaunchDimensions()
170 llvm::LLVMContext& llvm_context = llvm_module->getContext(); in UpdateLaunchDimensions()
566 llvm::Module* module = ir_emitter_context_->llvm_module(); in BuildKernelPrototype()
763 ir_emitter_context_->llvm_module()->getContext(), content); in EmitConstant()
779 llvm_ir::GetGlobalMemoryAddressSpace(*ir_emitter_context_->llvm_module()); in EmitConstant()
789 ir_emitter_context_->llvm_module()->getGlobalList().push_back( in EmitConstant()
927 ir_emitter_context_->llvm_module()); in EmitPadToStaticFromMlir()
1050 ir_emitter_context_->llvm_module()); in EmitSliceToDynamicFromMlir()
[all …]
Dnvptx_compiler.cc302 llvm::Module* llvm_module, in CompileTargetBinary() argument
329 TF_ASSIGN_OR_RETURN(ptx, nvptx::CompileToPtx(llvm_module, gpu_version, in CompileTargetBinary()
Dnvptx_compiler.h54 const HloModuleConfig& module_config, llvm::Module* llvm_module,
Dir_emitter.cc82 module_(ir_emitter_context->llvm_module()), in IrEmitter()
132 *ir_emitter_context_->llvm_module()); in EmitConstants()
144 ir_emitter_context_->llvm_module()->getGlobalList().push_back( in EmitConstants()
Dir_emitter_unnested.h356 shape, ir_emitter_context_->llvm_module()->getDataLayout()); in ByteSizeOf()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/tests/
Dmlir_gpu_test_base.cc44 auto llvm_module = absl::make_unique<llvm::Module>("", llvm_context); in CompileMlirModule() local
46 llvm_module->setTargetTriple(amdgpu::kTargetTriple); in CompileMlirModule()
47 llvm_module->setDataLayout(amdgpu::kDataLayout); in CompileMlirModule()
49 llvm_module->setTargetTriple(nvptx::kTargetTriple); in CompileMlirModule()
50 llvm_module->setDataLayout(nvptx::kDataLayout); in CompileMlirModule()
71 llvm_module.get()); in CompileMlirModule()
78 std::move(llvm_module), &ir_emitter_context); in CompileMlirModule()
Dhlo_to_llvm_ir.cc58 TF_ASSIGN_OR_RETURN(std::unique_ptr<llvm::Module> llvm_module, in CompileAndPrintLlvmIr()
66 llvm_module->print(llvm::outs(), nullptr); in CompileAndPrintLlvmIr()
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/
Dtf_to_kernel.cc77 std::unique_ptr<llvm::Module> llvm_module = in EmitToBinary() local
80 auto target_machine = GetTargetMachine(llvm_module.get()); in EmitToBinary()
81 llvm_module->setDataLayout(target_machine->createDataLayout()); in EmitToBinary()
86 target_machine.get())(llvm_module.get())) { in EmitToBinary()
97 llvm::Triple(llvm_module->getTargetTriple()))); in EmitToBinary()
103 codegen_passes.run(*llvm_module); in EmitToBinary()
/external/llvm-project/lldb/source/Plugins/ExpressionParser/Clang/
DIRForTarget.h104 bool runOnModule(llvm::Module &llvm_module) override;
/external/tensorflow/tensorflow/compiler/xla/tests/
Dllvm_compiler_test.cc61 const HloModuleConfig& module_config, llvm::Module* llvm_module, in CompileTargetBinary() argument
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dllvm_util.cc584 const llvm::Module& llvm_module, bool optimized, in DumpIrIfEnabled() argument
598 DumpModuleToString(llvm_module)); in DumpIrIfEnabled()
606 DumpModuleToString(*DropConstantInitializers(llvm_module))); in DumpIrIfEnabled()
Dllvm_util.h278 const llvm::Module& llvm_module, bool optimized,
/external/mesa3d/src/amd/vulkan/
Dradv_nir_to_llvm.c4125 LLVMModuleRef llvm_module, in ac_compile_llvm_module() argument
4137 ac_dump_module(llvm_module); in ac_compile_llvm_module()
4142 char *llvm_ir = LLVMPrintModuleToString(llvm_module); in ac_compile_llvm_module()
4147 int v = radv_llvm_compile(llvm_module, &elf_buffer, &elf_size, ac_llvm); in ac_compile_llvm_module()
4152 LLVMContextRef ctx = LLVMGetModuleContext(llvm_module); in ac_compile_llvm_module()
4153 LLVMDisposeModule(llvm_module); in ac_compile_llvm_module()
4182 LLVMModuleRef llvm_module; in radv_compile_nir_shader() local
4184 llvm_module = ac_translate_nir_to_llvm(ac_llvm, nir, nir_count, args); in radv_compile_nir_shader()
4186 ac_compile_llvm_module(ac_llvm, llvm_module, rbinary, in radv_compile_nir_shader()

12