| /* |
| * Copyright (C) 2016 The Android Open Source Project |
| * |
| * Licensed under the Apache License, Version 2.0 (the "License"); |
| * you may not use this file except in compliance with the License. |
| * You may obtain a copy of the License at |
| * |
| * http://www.apache.org/licenses/LICENSE-2.0 |
| * |
| * Unless required by applicable law or agreed to in writing, software |
| * distributed under the License is distributed on an "AS IS" BASIS, |
| * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. |
| * See the License for the specific language governing permissions and |
| * limitations under the License. |
| */ |
| |
| #ifndef ART_COMPILER_OPTIMIZING_CODEGEN_TEST_UTILS_H_ |
| #define ART_COMPILER_OPTIMIZING_CODEGEN_TEST_UTILS_H_ |
| |
| #include "arch/arm/instruction_set_features_arm.h" |
| #include "arch/arm/registers_arm.h" |
| #include "arch/arm64/instruction_set_features_arm64.h" |
| #include "arch/instruction_set.h" |
| #include "arch/mips/instruction_set_features_mips.h" |
| #include "arch/mips/registers_mips.h" |
| #include "arch/mips64/instruction_set_features_mips64.h" |
| #include "arch/mips64/registers_mips64.h" |
| #include "arch/x86/instruction_set_features_x86.h" |
| #include "arch/x86/registers_x86.h" |
| #include "arch/x86_64/instruction_set_features_x86_64.h" |
| #include "code_simulator_container.h" |
| #include "common_compiler_test.h" |
| #include "graph_checker.h" |
| #include "prepare_for_register_allocation.h" |
| #include "ssa_liveness_analysis.h" |
| |
| #ifdef ART_ENABLE_CODEGEN_arm |
| #include "code_generator_arm.h" |
| #include "code_generator_arm_vixl.h" |
| #endif |
| |
| #ifdef ART_ENABLE_CODEGEN_arm64 |
| #include "code_generator_arm64.h" |
| #endif |
| |
| #ifdef ART_ENABLE_CODEGEN_x86 |
| #include "code_generator_x86.h" |
| #endif |
| |
| #ifdef ART_ENABLE_CODEGEN_x86_64 |
| #include "code_generator_x86_64.h" |
| #endif |
| |
| #ifdef ART_ENABLE_CODEGEN_mips |
| #include "code_generator_mips.h" |
| #endif |
| |
| #ifdef ART_ENABLE_CODEGEN_mips64 |
| #include "code_generator_mips64.h" |
| #endif |
| |
| namespace art { |
| |
| typedef CodeGenerator* (*CreateCodegenFn)(HGraph*, const CompilerOptions&); |
| |
| class CodegenTargetConfig { |
| public: |
| CodegenTargetConfig(InstructionSet isa, CreateCodegenFn create_codegen) |
| : isa_(isa), create_codegen_(create_codegen) { |
| } |
| InstructionSet GetInstructionSet() const { return isa_; } |
| CodeGenerator* CreateCodeGenerator(HGraph* graph, const CompilerOptions& compiler_options) { |
| return create_codegen_(graph, compiler_options); |
| } |
| |
| private: |
| CodegenTargetConfig() {} |
| InstructionSet isa_; |
| CreateCodegenFn create_codegen_; |
| }; |
| |
| #ifdef ART_ENABLE_CODEGEN_arm |
| // Provide our own codegen, that ensures the C calling conventions |
| // are preserved. Currently, ART and C do not match as R4 is caller-save |
| // in ART, and callee-save in C. Alternatively, we could use or write |
| // the stub that saves and restores all registers, but it is easier |
| // to just overwrite the code generator. |
| class TestCodeGeneratorARM : public arm::CodeGeneratorARM { |
| public: |
| TestCodeGeneratorARM(HGraph* graph, |
| const ArmInstructionSetFeatures& isa_features, |
| const CompilerOptions& compiler_options) |
| : arm::CodeGeneratorARM(graph, isa_features, compiler_options) { |
| AddAllocatedRegister(Location::RegisterLocation(arm::R6)); |
| AddAllocatedRegister(Location::RegisterLocation(arm::R7)); |
| } |
| |
| void SetupBlockedRegisters() const OVERRIDE { |
| arm::CodeGeneratorARM::SetupBlockedRegisters(); |
| blocked_core_registers_[arm::R4] = true; |
| blocked_core_registers_[arm::R6] = false; |
| blocked_core_registers_[arm::R7] = false; |
| } |
| }; |
| |
| // A way to test the VIXL32-based code generator on ARM. This will replace |
| // TestCodeGeneratorARM when the VIXL32-based backend replaces the existing one. |
| class TestCodeGeneratorARMVIXL : public arm::CodeGeneratorARMVIXL { |
| public: |
| TestCodeGeneratorARMVIXL(HGraph* graph, |
| const ArmInstructionSetFeatures& isa_features, |
| const CompilerOptions& compiler_options) |
| : arm::CodeGeneratorARMVIXL(graph, isa_features, compiler_options) { |
| AddAllocatedRegister(Location::RegisterLocation(arm::R6)); |
| AddAllocatedRegister(Location::RegisterLocation(arm::R7)); |
| } |
| |
| void SetupBlockedRegisters() const OVERRIDE { |
| arm::CodeGeneratorARMVIXL::SetupBlockedRegisters(); |
| blocked_core_registers_[arm::R4] = true; |
| blocked_core_registers_[arm::R6] = false; |
| blocked_core_registers_[arm::R7] = false; |
| } |
| }; |
| #endif |
| |
| #ifdef ART_ENABLE_CODEGEN_x86 |
| class TestCodeGeneratorX86 : public x86::CodeGeneratorX86 { |
| public: |
| TestCodeGeneratorX86(HGraph* graph, |
| const X86InstructionSetFeatures& isa_features, |
| const CompilerOptions& compiler_options) |
| : x86::CodeGeneratorX86(graph, isa_features, compiler_options) { |
| // Save edi, we need it for getting enough registers for long multiplication. |
| AddAllocatedRegister(Location::RegisterLocation(x86::EDI)); |
| } |
| |
| void SetupBlockedRegisters() const OVERRIDE { |
| x86::CodeGeneratorX86::SetupBlockedRegisters(); |
| // ebx is a callee-save register in C, but caller-save for ART. |
| blocked_core_registers_[x86::EBX] = true; |
| |
| // Make edi available. |
| blocked_core_registers_[x86::EDI] = false; |
| } |
| }; |
| #endif |
| |
| class InternalCodeAllocator : public CodeAllocator { |
| public: |
| InternalCodeAllocator() : size_(0) { } |
| |
| virtual uint8_t* Allocate(size_t size) { |
| size_ = size; |
| memory_.reset(new uint8_t[size]); |
| return memory_.get(); |
| } |
| |
| size_t GetSize() const { return size_; } |
| uint8_t* GetMemory() const { return memory_.get(); } |
| |
| private: |
| size_t size_; |
| std::unique_ptr<uint8_t[]> memory_; |
| |
| DISALLOW_COPY_AND_ASSIGN(InternalCodeAllocator); |
| }; |
| |
| static bool CanExecuteOnHardware(InstructionSet target_isa) { |
| return (target_isa == kRuntimeISA) |
| // Handle the special case of ARM, with two instructions sets (ARM32 and Thumb-2). |
| || (kRuntimeISA == kArm && target_isa == kThumb2); |
| } |
| |
| static bool CanExecute(InstructionSet target_isa) { |
| CodeSimulatorContainer simulator(target_isa); |
| return CanExecuteOnHardware(target_isa) || simulator.CanSimulate(); |
| } |
| |
| template <typename Expected> |
| inline static Expected SimulatorExecute(CodeSimulator* simulator, Expected (*f)()); |
| |
| template <> |
| inline bool SimulatorExecute<bool>(CodeSimulator* simulator, bool (*f)()) { |
| simulator->RunFrom(reinterpret_cast<intptr_t>(f)); |
| return simulator->GetCReturnBool(); |
| } |
| |
| template <> |
| inline int32_t SimulatorExecute<int32_t>(CodeSimulator* simulator, int32_t (*f)()) { |
| simulator->RunFrom(reinterpret_cast<intptr_t>(f)); |
| return simulator->GetCReturnInt32(); |
| } |
| |
| template <> |
| inline int64_t SimulatorExecute<int64_t>(CodeSimulator* simulator, int64_t (*f)()) { |
| simulator->RunFrom(reinterpret_cast<intptr_t>(f)); |
| return simulator->GetCReturnInt64(); |
| } |
| |
| template <typename Expected> |
| static void VerifyGeneratedCode(InstructionSet target_isa, |
| Expected (*f)(), |
| bool has_result, |
| Expected expected) { |
| ASSERT_TRUE(CanExecute(target_isa)) << "Target isa is not executable."; |
| |
| // Verify on simulator. |
| CodeSimulatorContainer simulator(target_isa); |
| if (simulator.CanSimulate()) { |
| Expected result = SimulatorExecute<Expected>(simulator.Get(), f); |
| if (has_result) { |
| ASSERT_EQ(expected, result); |
| } |
| } |
| |
| // Verify on hardware. |
| if (CanExecuteOnHardware(target_isa)) { |
| Expected result = f(); |
| if (has_result) { |
| ASSERT_EQ(expected, result); |
| } |
| } |
| } |
| |
| template <typename Expected> |
| static void Run(const InternalCodeAllocator& allocator, |
| const CodeGenerator& codegen, |
| bool has_result, |
| Expected expected) { |
| InstructionSet target_isa = codegen.GetInstructionSet(); |
| |
| typedef Expected (*fptr)(); |
| CommonCompilerTest::MakeExecutable(allocator.GetMemory(), allocator.GetSize()); |
| fptr f = reinterpret_cast<fptr>(allocator.GetMemory()); |
| if (target_isa == kThumb2) { |
| // For thumb we need the bottom bit set. |
| f = reinterpret_cast<fptr>(reinterpret_cast<uintptr_t>(f) + 1); |
| } |
| VerifyGeneratedCode(target_isa, f, has_result, expected); |
| } |
| |
| static void ValidateGraph(HGraph* graph) { |
| GraphChecker graph_checker(graph); |
| graph_checker.Run(); |
| if (!graph_checker.IsValid()) { |
| for (const auto& error : graph_checker.GetErrors()) { |
| std::cout << error << std::endl; |
| } |
| } |
| ASSERT_TRUE(graph_checker.IsValid()); |
| } |
| |
| template <typename Expected> |
| static void RunCodeNoCheck(CodeGenerator* codegen, |
| HGraph* graph, |
| const std::function<void(HGraph*)>& hook_before_codegen, |
| bool has_result, |
| Expected expected) { |
| SsaLivenessAnalysis liveness(graph, codegen); |
| PrepareForRegisterAllocation(graph).Run(); |
| liveness.Analyze(); |
| RegisterAllocator::Create(graph->GetArena(), codegen, liveness)->AllocateRegisters(); |
| hook_before_codegen(graph); |
| InternalCodeAllocator allocator; |
| codegen->Compile(&allocator); |
| Run(allocator, *codegen, has_result, expected); |
| } |
| |
| template <typename Expected> |
| static void RunCode(CodeGenerator* codegen, |
| HGraph* graph, |
| std::function<void(HGraph*)> hook_before_codegen, |
| bool has_result, |
| Expected expected) { |
| ValidateGraph(graph); |
| RunCodeNoCheck(codegen, graph, hook_before_codegen, has_result, expected); |
| } |
| |
| template <typename Expected> |
| static void RunCode(CodegenTargetConfig target_config, |
| HGraph* graph, |
| std::function<void(HGraph*)> hook_before_codegen, |
| bool has_result, |
| Expected expected) { |
| CompilerOptions compiler_options; |
| std::unique_ptr<CodeGenerator> codegen(target_config.CreateCodeGenerator(graph, compiler_options)); |
| RunCode(codegen.get(), graph, hook_before_codegen, has_result, expected); |
| } |
| |
| #ifdef ART_ENABLE_CODEGEN_arm |
| CodeGenerator* create_codegen_arm(HGraph* graph, const CompilerOptions& compiler_options) { |
| std::unique_ptr<const ArmInstructionSetFeatures> features_arm( |
| ArmInstructionSetFeatures::FromCppDefines()); |
| return new (graph->GetArena()) TestCodeGeneratorARM(graph, |
| *features_arm.get(), |
| compiler_options); |
| } |
| |
| CodeGenerator* create_codegen_arm_vixl32(HGraph* graph, const CompilerOptions& compiler_options) { |
| std::unique_ptr<const ArmInstructionSetFeatures> features_arm( |
| ArmInstructionSetFeatures::FromCppDefines()); |
| return new (graph->GetArena()) |
| TestCodeGeneratorARMVIXL(graph, *features_arm.get(), compiler_options); |
| } |
| #endif |
| |
| #ifdef ART_ENABLE_CODEGEN_arm64 |
| CodeGenerator* create_codegen_arm64(HGraph* graph, const CompilerOptions& compiler_options) { |
| std::unique_ptr<const Arm64InstructionSetFeatures> features_arm64( |
| Arm64InstructionSetFeatures::FromCppDefines()); |
| return new (graph->GetArena()) arm64::CodeGeneratorARM64(graph, |
| *features_arm64.get(), |
| compiler_options); |
| } |
| #endif |
| |
| #ifdef ART_ENABLE_CODEGEN_x86 |
| CodeGenerator* create_codegen_x86(HGraph* graph, const CompilerOptions& compiler_options) { |
| std::unique_ptr<const X86InstructionSetFeatures> features_x86( |
| X86InstructionSetFeatures::FromCppDefines()); |
| return new (graph->GetArena()) TestCodeGeneratorX86(graph, *features_x86.get(), compiler_options); |
| } |
| #endif |
| |
| #ifdef ART_ENABLE_CODEGEN_x86_64 |
| CodeGenerator* create_codegen_x86_64(HGraph* graph, const CompilerOptions& compiler_options) { |
| std::unique_ptr<const X86_64InstructionSetFeatures> features_x86_64( |
| X86_64InstructionSetFeatures::FromCppDefines()); |
| return new (graph->GetArena()) |
| x86_64::CodeGeneratorX86_64(graph, *features_x86_64.get(), compiler_options); |
| } |
| #endif |
| |
| #ifdef ART_ENABLE_CODEGEN_mips |
| CodeGenerator* create_codegen_mips(HGraph* graph, const CompilerOptions& compiler_options) { |
| std::unique_ptr<const MipsInstructionSetFeatures> features_mips( |
| MipsInstructionSetFeatures::FromCppDefines()); |
| return new (graph->GetArena()) |
| mips::CodeGeneratorMIPS(graph, *features_mips.get(), compiler_options); |
| } |
| #endif |
| |
| #ifdef ART_ENABLE_CODEGEN_mips64 |
| CodeGenerator* create_codegen_mips64(HGraph* graph, const CompilerOptions& compiler_options) { |
| std::unique_ptr<const Mips64InstructionSetFeatures> features_mips64( |
| Mips64InstructionSetFeatures::FromCppDefines()); |
| return new (graph->GetArena()) |
| mips64::CodeGeneratorMIPS64(graph, *features_mips64.get(), compiler_options); |
| } |
| #endif |
| |
| } // namespace art |
| |
| #endif // ART_COMPILER_OPTIMIZING_CODEGEN_TEST_UTILS_H_ |