| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.cc | 456 const llvm::Module& llvm_module) { in GetIRModuleHooks() argument
460 user_hook(llvm_module); in GetIRModuleHooks()
462 llvm_ir::DumpIrIfEnabled(*hlo_module_ptr, llvm_module, optimized); in GetIRModuleHooks()
464 return {[hook](const llvm::Module& llvm_module) { in GetIRModuleHooks()
465 return hook(/*optimized=*/false, llvm_module); in GetIRModuleHooks()
467 [hook](const llvm::Module& llvm_module) { in GetIRModuleHooks()
468 return hook(/*optimized=*/true, llvm_module); in GetIRModuleHooks()
472 Status VerifyLlvmModule(const llvm::Module& llvm_module) { in VerifyLlvmModule() argument
479 TF_RET_CHECK(!llvm::verifyModule(llvm_module, &err_stream)) in VerifyLlvmModule()
603 auto llvm_module = in RunBackend() local
[all …]
|
| D | ir_function.cc | 30 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()
|
| D | ir_function.h | 56 const HloModuleConfig& module_config, llvm::Module* llvm_module,
|
| D | ir_emitter.h | 77 llvm::Module* llvm_module,
|
| D | ir_emitter.cc | 85 llvm::Module* llvm_module, in IrEmitter() argument
91 module_(llvm_module), in IrEmitter()
92 arch_type_(llvm::Triple(llvm_module->getTargetTriple()).getArch()), in IrEmitter()
93 b_(llvm_module->getContext()), in IrEmitter()
96 alias_analysis_(hlo_module, assignment, &llvm_module->getContext()), in IrEmitter()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | ir_emitter_context.h | 36 llvm::Module* llvm_module) in IrEmitterContext() argument
40 llvm_module_(llvm_module) {} in IrEmitterContext()
53 llvm::Module* llvm_module() { return llvm_module_; } in llvm_module() function
|
| D | nvptx_compiler.cc | 651 llvm::Module llvm_module(module->name().c_str(), llvm_context); in RunBackend() local
653 llvm_module.setTargetTriple(kTargetTriple); in RunBackend()
654 llvm_module.setDataLayout(kDataLayout); in RunBackend()
683 &llvm_module); in RunBackend()
697 user_pre_optimization_hook_(llvm_module); in RunBackend()
703 ir_module_string_before_opt = llvm_ir::DumpModuleToString(llvm_module); in RunBackend()
706 llvm_ir::DumpIrIfEnabled(*module, llvm_module, /*optimized=*/false); in RunBackend()
716 TF_RET_CHECK(!llvm::verifyModule(llvm_module, &err_stream)) in RunBackend()
749 TF_ASSIGN_OR_RETURN(ptx, CompileToPtx(&llvm_module, {cc_major, cc_minor}, in RunBackend()
753 llvm_ir::DumpIrIfEnabled(*module, llvm_module, /*optimized=*/true); in RunBackend()
[all …]
|
| D | hlo_to_ir_bindings.h | 40 llvm::IRBuilder<>* b, llvm::Module* llvm_module, in HloToIrBindings() argument
45 module_(llvm_module), in HloToIrBindings()
|
| D | ir_emitter_nested.cc | 69 root_shape, ir_emitter_context_->llvm_module()->getDataLayout()); in EmitBasePointersForNestedComputation()
83 ir_emitter_context_->llvm_module()); // The parent LLVM module. in EmitBasePointersForNestedComputation()
|
| D | ir_emitter_unnested.cc | 148 llvm::Module* llvm_module) { in UpdateLaunchDimensions() argument
156 llvm_module->getOrInsertNamedMetadata("nvvm.annotations"); in UpdateLaunchDimensions()
158 llvm_module->getFunction(kernel_thunk->kernel_name().c_str()); in UpdateLaunchDimensions()
159 llvm::LLVMContext& llvm_context = llvm_module->getContext(); in UpdateLaunchDimensions()
197 llvm::Module* module = ir_emitter_context_->llvm_module(); in BuildKernelPrototype()
590 hlo_module_config_, ir_emitter_context_->llvm_module(), &b_, in HandleFusion()
611 hlo_module_config_, ir_emitter_context_->llvm_module(), &b_, in HandleFusion()
654 ir_emitter_context_->llvm_module(), in HandleFusion()
667 ir_emitter_context_->llvm_module()); in HandleFusion()
840 ir_emitter_context_->llvm_module()), in HandleSelectAndScatter()
[all …]
|
| D | ir_emitter.cc | 60 module_(ir_emitter_context->llvm_module()), in IrEmitter()
|
| /external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/ |
| D | llvm_util.cc | 586 const llvm::Module& llvm_module, bool optimized) { in DumpIrIfEnabled() argument
596 DumpModuleToString(llvm_module)); in DumpIrIfEnabled()
604 DumpModuleToString(*DropConstantInitializers(llvm_module))); in DumpIrIfEnabled()
|
| D | llvm_util.h | 283 const llvm::Module& llvm_module, bool optimized);
|
| /external/mesa3d/src/amd/common/ |
| D | ac_nir_to_llvm.c | 7031 LLVMModuleRef llvm_module, in ac_compile_llvm_module() argument
7039 ac_dump_module(llvm_module); in ac_compile_llvm_module()
7042 int v = ac_llvm_compile(llvm_module, binary, tm); in ac_compile_llvm_module()
7052 LLVMContextRef ctx = LLVMGetModuleContext(llvm_module); in ac_compile_llvm_module()
7053 LLVMDisposeModule(llvm_module); in ac_compile_llvm_module()
7161 LLVMModuleRef llvm_module = ac_translate_nir_to_llvm(tm, nir, nir_count, shader_info, in ac_compile_nir_shader() local
7164 …ac_compile_llvm_module(tm, llvm_module, binary, config, shader_info, nir[0]->info.stage, dump_shad… in ac_compile_nir_shader()
|