diff --git a/src/infinicore/ops/all/al_infiniop.cc b/src/infinicore/ops/all/all_infiniop.cc similarity index 82% rename from src/infinicore/ops/all/al_infiniop.cc rename to src/infinicore/ops/all/all_infiniop.cc index 094716ba8..3d24aca12 100644 --- a/src/infinicore/ops/all/al_infiniop.cc +++ b/src/infinicore/ops/all/all_infiniop.cc @@ -45,12 +45,7 @@ void calculate(Tensor output, Tensor input, std::vector dim, bool keepdi } static bool registered = []() { - All::dispatcher().registerDevice({Device::Type::CPU, - Device::Type::NVIDIA, - Device::Type::METAX, - Device::Type::MOORE, - Device::Type::ILUVATAR}, - &calculate, false); + All::dispatcher().registerAll(&calculate, false); return true; }(); diff --git a/src/infinicore/ops/sum/sum_infiniop.cc b/src/infinicore/ops/sum/sum_infiniop.cc index 9a696a9b5..700ab04fa 100644 --- a/src/infinicore/ops/sum/sum_infiniop.cc +++ b/src/infinicore/ops/sum/sum_infiniop.cc @@ -45,12 +45,7 @@ void calculate(Tensor output, Tensor input, std::vector dim, bool keepdi } static bool registered = []() { - Sum::dispatcher().registerDevice({Device::Type::CPU, - Device::Type::NVIDIA, - Device::Type::METAX, - Device::Type::MOORE, - Device::Type::ILUVATAR}, - &calculate, false); + Sum::dispatcher().registerAll(&calculate, false); return true; }(); diff --git a/src/infinicore/ops/topk/topk_infiniop.cc b/src/infinicore/ops/topk/topk_infiniop.cc index 5cc8d4d98..db7e0471f 100644 --- a/src/infinicore/ops/topk/topk_infiniop.cc +++ b/src/infinicore/ops/topk/topk_infiniop.cc @@ -45,12 +45,7 @@ void calculate(Tensor values_output, Tensor indices_output, Tensor input, size_t } static bool registered = []() { - TopK::dispatcher().registerDevice({Device::Type::CPU, - Device::Type::NVIDIA, - Device::Type::METAX, - Device::Type::MOORE, - Device::Type::ILUVATAR}, - &calculate, false); + TopK::dispatcher().registerAll(&calculate, false); return true; }(); diff --git a/src/infinicore/ops/var/var_infiniop.cc b/src/infinicore/ops/var/var_infiniop.cc index c74eb2628..9c162cec7 100644 --- a/src/infinicore/ops/var/var_infiniop.cc +++ b/src/infinicore/ops/var/var_infiniop.cc @@ -45,12 +45,7 @@ void calculate(Tensor var_output, Tensor input, std::vector dim, bool un } static bool registered = []() { - Var::dispatcher().registerDevice({Device::Type::CPU, - Device::Type::NVIDIA, - Device::Type::METAX, - Device::Type::MOORE, - Device::Type::ILUVATAR}, - &calculate, false); + Var::dispatcher().registerAll(&calculate, false); return true; }(); diff --git a/src/infinicore/ops/var_mean/var_mean_infiniop.cc b/src/infinicore/ops/var_mean/var_mean_infiniop.cc index 89332d074..5f74a3dbf 100644 --- a/src/infinicore/ops/var_mean/var_mean_infiniop.cc +++ b/src/infinicore/ops/var_mean/var_mean_infiniop.cc @@ -47,12 +47,7 @@ void calculate(Tensor var_output, Tensor mean_output, Tensor input, std::vector< } static bool registered = []() { - Var_Mean::dispatcher().registerDevice({Device::Type::CPU, - Device::Type::NVIDIA, - Device::Type::METAX, - Device::Type::MOORE, - Device::Type::ILUVATAR}, - &calculate, false); + Var_Mean::dispatcher().registerAll(&calculate, false); return true; }();