bugfix: add USE_NPU macro to support multiple platforms.

Author: XuZhang99

xllm/core/layers/rms_norm.h

@@ -14,16 +14,20 @@ limitations under the License.
 ==============================================================================*/
 
 #pragma once
+#if defined(USE_NPU)
 #ifdef TORCH_HIGHER_THAN_PTA6
 #include <torch_npu/csrc/core/npu/NPUFormat.h>
 #include <torch_npu/csrc/framework/OpCommand.h>
 #else
 #include <torch_npu/csrc/aten/NPUNativeFunctions.h>
 #include <torch_npu/csrc/framework/utils/OpPreparation.h>
 #endif
-
 #include <torch_npu/csrc/libs/init_npu.h>
 
+#include "npu/npu_rms_norm_impl.h"
+#include "pytorch/adapter/utils/utils.h"
+#endif
+
 #include <functional>
 
 #include "atb/atb_infer.h"
@@ -36,8 +40,6 @@ limitations under the License.
 #include "framework/model/model_input_params.h"
 #include "framework/state_dict/state_dict.h"
 #include "nlohmann/json.hpp"
-#include "npu/npu_rms_norm_impl.h"
-#include "pytorch/adapter/utils/utils.h"
 
 namespace xllm {