Skip to content

Commit

Permalink
Remove PI
Browse files Browse the repository at this point in the history
  • Loading branch information
maliasadi committed Sep 25, 2024
1 parent c607b6c commit 3db3c2c
Show file tree
Hide file tree
Showing 12 changed files with 11 additions and 2,024 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@ if (ENABLE_GATE_DISPATCHER AND UNIX AND (${CMAKE_SYSTEM_PROCESSOR} MATCHES "(AMD
target_include_directories(lq_gates_register_kernels_avx512 PRIVATE ${CMAKE_CURRENT_SOURCE_DIR})
set_target_properties(lq_gates_register_kernels_avx512 PROPERTIES POSITION_INDEPENDENT_CODE ON)

add_library(lq_gates_register_kernels_x64 STATIC RegisterKernels_x64.cpp cpu_kernels/GateImplementationsLM.cpp cpu_kernels/GateImplementationsPI.cpp GateIndices.cpp)
add_library(lq_gates_register_kernels_x64 STATIC RegisterKernels_x64.cpp cpu_kernels/GateImplementationsLM.cpp GateIndices.cpp)
target_link_libraries(lq_gates_register_kernels_x64 PRIVATE lightning_external_libs lightning_compile_options lightning_gates lightning_utils lightning_qubit_utils)
target_include_directories(lq_gates_register_kernels_x64 PRIVATE ${CMAKE_CURRENT_SOURCE_DIR})
set_target_properties(lq_gates_register_kernels_x64 PROPERTIES POSITION_INDEPENDENT_CODE ON)
Expand All @@ -53,7 +53,7 @@ else()
lightning_gates)
set_target_properties(lq_gates_kernel_map PROPERTIES POSITION_INDEPENDENT_CODE ON)

add_library(lq_gates_register_kernels_default STATIC RegisterKernels_Default.cpp cpu_kernels/GateImplementationsLM.cpp cpu_kernels/GateImplementationsPI.cpp GateIndices.cpp)
add_library(lq_gates_register_kernels_default STATIC RegisterKernels_Default.cpp cpu_kernels/GateImplementationsLM.cpp GateIndices.cpp)
target_include_directories(lq_gates_register_kernels_default PRIVATE ${CMAKE_CURRENT_SOURCE_DIR})
target_link_libraries(lq_gates_register_kernels_default PRIVATE lightning_gates
lightning_external_libs
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -241,11 +241,11 @@ class OperationKernelMap {
OperationKernelMap()
: allowed_kernels_{
// LCOV_EXCL_START
{CPUMemoryModel::Unaligned, {KernelType::LM, KernelType::PI}},
{CPUMemoryModel::Unaligned, {KernelType::LM}},
{CPUMemoryModel::Aligned256,
{KernelType::LM, KernelType::PI, KernelType::AVX2}},
{KernelType::LM, KernelType::AVX2}},
{CPUMemoryModel::Aligned512,
{KernelType::LM, KernelType::PI, KernelType::AVX2,
{KernelType::LM, KernelType::AVX2,
KernelType::AVX512}},
// LCOV_EXCL_STOP
} {}
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -23,5 +23,5 @@ namespace Pennylane::Gates {
/**
* @brief Define kernel id for each implementation.
*/
enum class KernelType { PI, LM, AVX2, AVX512, None };
enum class KernelType { LM, AVX2, AVX512, None };
} // namespace Pennylane::Gates
Original file line number Diff line number Diff line change
Expand Up @@ -18,18 +18,15 @@
#include "DynamicDispatcher.hpp"
#include "RegisterKernel.hpp"
#include "cpu_kernels/GateImplementationsLM.hpp"
#include "cpu_kernels/GateImplementationsPI.hpp"

namespace Pennylane::LightningQubit::Internal {
int registerAllAvailableKernels_Float() {
registerKernel<float, float, Gates::GateImplementationsLM>();
registerKernel<float, float, Gates::GateImplementationsPI>();
return 1;
}

int registerAllAvailableKernels_Double() {
registerKernel<double, double, Gates::GateImplementationsLM>();
registerKernel<double, double, Gates::GateImplementationsPI>();
return 1;
}
} // namespace Pennylane::LightningQubit::Internal
Original file line number Diff line number Diff line change
Expand Up @@ -20,13 +20,11 @@
#include "RegisterKernel.hpp"
#include "RuntimeInfo.hpp"
#include "cpu_kernels/GateImplementationsLM.hpp"
#include "cpu_kernels/GateImplementationsPI.hpp"

namespace Pennylane::LightningQubit::Internal {
int registerAllAvailableKernels_Float() {
using Pennylane::Util::RuntimeInfo;
registerKernel<float, float, Gates::GateImplementationsLM>();
registerKernel<float, float, Gates::GateImplementationsPI>();

if (RuntimeInfo::AVX2() && RuntimeInfo::FMA()) {
registerKernelsAVX2_Float();
Expand All @@ -40,7 +38,6 @@ int registerAllAvailableKernels_Float() {
int registerAllAvailableKernels_Double() {
using Pennylane::Util::RuntimeInfo;
registerKernel<double, double, Gates::GateImplementationsLM>();
registerKernel<double, double, Gates::GateImplementationsPI>();

if (RuntimeInfo::AVX2() && RuntimeInfo::FMA()) {
registerKernelsAVX2_Double();
Expand Down
Loading

0 comments on commit 3db3c2c

Please sign in to comment.