diff --git a/docs/aten-operators.md b/docs/aten-operators.md index 766508897..75404888d 100644 --- a/docs/aten-operators.md +++ b/docs/aten-operators.md @@ -323,9 +323,7 @@ template class Operator; template class Operator; template class Operator; template class Operator; -template class Operator; template class Operator; -template class Operator; } // namespace infini::ops ``` diff --git a/scripts/generate_torch_ops.py b/scripts/generate_torch_ops.py index 64a93f9cf..9e45a5f4e 100644 --- a/scripts/generate_torch_ops.py +++ b/scripts/generate_torch_ops.py @@ -71,9 +71,7 @@ "kMetax", "kMoore", "kIluvatar", - "kKunlun", "kHygon", - "kQy", ) # YAML scalar-type tokens → C++ types. Reference types (e.g. `const Scalar&`) diff --git a/scripts/generate_wrappers.py b/scripts/generate_wrappers.py index 9891ad7d9..e2630ad26 100644 --- a/scripts/generate_wrappers.py +++ b/scripts/generate_wrappers.py @@ -765,9 +765,7 @@ def _generate_source(operator): {{INFINI_DEVICE_METAX, infini::ops::Device::Type::kMetax}}, {{INFINI_DEVICE_MOORE, infini::ops::Device::Type::kMoore}}, {{INFINI_DEVICE_ILUVATAR, infini::ops::Device::Type::kIluvatar}}, - {{INFINI_DEVICE_KUNLUN, infini::ops::Device::Type::kKunlun}}, - {{INFINI_DEVICE_HYGON, infini::ops::Device::Type::kHygon}}, - {{INFINI_DEVICE_QY, infini::ops::Device::Type::kQy}}}}}}}}; + {{INFINI_DEVICE_HYGON, infini::ops::Device::Type::kHygon}}}}}}}}; return kInfiniDeviceToDeviceType.at(device); }} @@ -1498,7 +1496,7 @@ def _dispatch_gen_batch_size(): nargs="+", default="cpu", type=str, - help="Devices to use. Please pick from `cpu`, `nvidia`, `cambricon`, `ascend`, `metax`, `moore`, `iluvatar`, `kunlun`, `hygon`, and `qy`. (default: `cpu`)", + help="Devices to use. Please pick from `cpu`, `nvidia`, `cambricon`, `ascend`, `metax`, `moore`, `iluvatar`, and `hygon`. (default: `cpu`)", ) parser.add_argument( diff --git a/src/device.h b/src/device.h index 41a3af3b1..59bbeb9c3 100644 --- a/src/device.h +++ b/src/device.h @@ -15,8 +15,7 @@ using DeviceEnabled = infini::rt::DeviceEnabled; using AllDeviceTypes = List; + Device::Type::kIluvatar, Device::Type::kHygon>; template struct ActiveDevicesImpl { diff --git a/src/pybind11_utils.h b/src/pybind11_utils.h index 6610e4ae1..1571e2981 100644 --- a/src/pybind11_utils.h +++ b/src/pybind11_utils.h @@ -96,9 +96,7 @@ inline std::optional TryDeviceTypeFromString( {"metax", Device::Type::kMetax}, {"moore", Device::Type::kMoore}, {"iluvatar", Device::Type::kIluvatar}, - {"kunlun", Device::Type::kKunlun}, {"hygon", Device::Type::kHygon}, - {"qy", Device::Type::kQy}, }; auto platform_it{kPlatformNames.find(name)}; diff --git a/src/torch/device_.h b/src/torch/device_.h index f83fc8a22..dfd3cae6b 100644 --- a/src/torch/device_.h +++ b/src/torch/device_.h @@ -45,21 +45,11 @@ struct TorchDeviceName { static constexpr std::string_view kValue{"cuda"}; }; -template <> -struct TorchDeviceName { - static constexpr std::string_view kValue{"cuda"}; -}; - template <> struct TorchDeviceName { static constexpr std::string_view kValue{"cuda"}; }; -template <> -struct TorchDeviceName { - static constexpr std::string_view kValue{"cuda"}; -}; - } // namespace infini::ops::detail #endif diff --git a/src/torch/ops/add/add.cc b/src/torch/ops/add/add.cc index 2108e652c..8e6530d78 100644 --- a/src/torch/ops/add/add.cc +++ b/src/torch/ops/add/add.cc @@ -31,8 +31,6 @@ template class Operator; template class Operator; template class Operator; template class Operator; -template class Operator; template class Operator; -template class Operator; } // namespace infini::ops diff --git a/src/torch/ops/gemm/gemm.cc b/src/torch/ops/gemm/gemm.cc index 8cd087dce..6b5ea6652 100644 --- a/src/torch/ops/gemm/gemm.cc +++ b/src/torch/ops/gemm/gemm.cc @@ -73,8 +73,6 @@ template class Operator; template class Operator; template class Operator; template class Operator; -template class Operator; template class Operator; -template class Operator; } // namespace infini::ops