Add 'projects/rdc/' from commit '5ae7eeb3550d4cb14cbc31d3022e545b054f1ad1'

git-subtree-dir: projects/rdc git-subtree-mainline: a68afa42a1 git-subtree-split: 5ae7eeb355
2025-07-22 22:52:37 +00:00
parent a68afa42a1 5ae7eeb355
commit c2312be1a2
393 changed files with 49849 additions and 0 deletions
@@ -0,0 +1,46 @@
+/*
+Copyright (c) 2019 - present Advanced Micro Devices, Inc. All rights reserved.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.
+*/
+#ifndef SERVER_INCLUDE_RDC_RDC_ADMIN_SERVICE_H_
+#define SERVER_INCLUDE_RDC_RDC_ADMIN_SERVICE_H_
+
+#include "amd_smi/amdsmi.h"
+#include "rdc.grpc.pb.h"  // NOLINT
+#include "rdc/rdc_admin_service.h"
+
+namespace amd {
+namespace rdc {
+
+class RDCAdminServiceImpl final : public ::rdc::RdcAdmin::Service {
+ public:
+  RDCAdminServiceImpl();
+  ~RDCAdminServiceImpl();
+  ::grpc::Status VerifyConnection(::grpc::ServerContext* context,
+                                  const ::rdc::VerifyConnectionRequest* request,
+                                  ::rdc::VerifyConnectionResponse* reply) override;
+
+ private:
+};
+
+}  // namespace rdc
+}  // namespace amd
+
+#endif  // SERVER_INCLUDE_RDC_RDC_ADMIN_SERVICE_H_
@@ -0,0 +1,213 @@
+/*
+Copyright (c) 2020 - present Advanced Micro Devices, Inc. All rights reserved.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.
+*/
+#ifndef SERVER_INCLUDE_RDC_RDC_API_SERVICE_H_
+#define SERVER_INCLUDE_RDC_RDC_API_SERVICE_H_
+
+#include <grpcpp/server_context.h>
+
+#include <thread>
+
+#include "rdc.grpc.pb.h"  // NOLINT
+#include "rdc/rdc.h"
+
+namespace amd {
+namespace rdc {
+
+class RdcAPIServiceImpl final : public ::rdc::RdcAPI::Service {
+ public:
+  RdcAPIServiceImpl();
+  ~RdcAPIServiceImpl();
+
+  rdc_status_t Initialize(uint64_t rdcd_init_flags = 0);
+  void Shutdown();
+
+  ::grpc::Status GetAllDevices(::grpc::ServerContext* context, const ::rdc::Empty* request,
+                               ::rdc::GetAllDevicesResponse* reply) override;
+
+  ::grpc::Status GetDeviceAttributes(::grpc::ServerContext* context,
+                                     const ::rdc::GetDeviceAttributesRequest* request,
+                                     ::rdc::GetDeviceAttributesResponse* reply) override;
+
+  ::grpc::Status GetComponentVersion(::grpc::ServerContext* context,
+                                     const ::rdc::GetComponentVersionRequest* request,
+                                     ::rdc::GetComponentVersionResponse* reply) override;
+
+  ::grpc::Status CreateGpuGroup(::grpc::ServerContext* context,
+                                const ::rdc::CreateGpuGroupRequest* request,
+                                ::rdc::CreateGpuGroupResponse* reply) override;
+
+  ::grpc::Status AddToGpuGroup(::grpc::ServerContext* context,
+                               const ::rdc::AddToGpuGroupRequest* request,
+                               ::rdc::AddToGpuGroupResponse* reply) override;
+
+  ::grpc::Status GetGpuGroupInfo(::grpc::ServerContext* context,
+                                 const ::rdc::GetGpuGroupInfoRequest* request,
+                                 ::rdc::GetGpuGroupInfoResponse* reply) override;
+
+  ::grpc::Status GetGroupAllIds(::grpc::ServerContext* context, const ::rdc::Empty* request,
+                                ::rdc::GetGroupAllIdsResponse* reply) override;
+
+  ::grpc::Status DestroyGpuGroup(::grpc::ServerContext* context,
+                                 const ::rdc::DestroyGpuGroupRequest* request,
+                                 ::rdc::DestroyGpuGroupResponse* reply) override;
+
+  ::grpc::Status CreateFieldGroup(::grpc::ServerContext* context,
+                                  const ::rdc::CreateFieldGroupRequest* request,
+                                  ::rdc::CreateFieldGroupResponse* reply) override;
+
+  ::grpc::Status GetFieldGroupInfo(::grpc::ServerContext* context,
+                                   const ::rdc::GetFieldGroupInfoRequest* request,
+                                   ::rdc::GetFieldGroupInfoResponse* reply) override;
+
+  ::grpc::Status GetFieldGroupAllIds(::grpc::ServerContext* context, const ::rdc::Empty* request,
+                                     ::rdc::GetFieldGroupAllIdsResponse* reply) override;
+
+  ::grpc::Status DestroyFieldGroup(::grpc::ServerContext* context,
+                                   const ::rdc::DestroyFieldGroupRequest* request,
+                                   ::rdc::DestroyFieldGroupResponse* reply) override;
+
+  ::grpc::Status WatchFields(::grpc::ServerContext* context,
+                             const ::rdc::WatchFieldsRequest* request,
+                             ::rdc::WatchFieldsResponse* reply) override;
+
+  ::grpc::Status GetLatestFieldValue(::grpc::ServerContext* context,
+                                     const ::rdc::GetLatestFieldValueRequest* request,
+                                     ::rdc::GetLatestFieldValueResponse* reply) override;
+
+  ::grpc::Status GetFieldSince(::grpc::ServerContext* context,
+                               const ::rdc::GetFieldSinceRequest* request,
+                               ::rdc::GetFieldSinceResponse* reply) override;
+
+  ::grpc::Status UnWatchFields(::grpc::ServerContext* context,
+                               const ::rdc::UnWatchFieldsRequest* request,
+                               ::rdc::UnWatchFieldsResponse* reply) override;
+
+  ::grpc::Status UpdateAllFields(::grpc::ServerContext* context,
+                                 const ::rdc::UpdateAllFieldsRequest* request,
+                                 ::rdc::UpdateAllFieldsResponse* reply) override;
+
+  ::grpc::Status StartJobStats(::grpc::ServerContext* context,
+                               const ::rdc::StartJobStatsRequest* request,
+                               ::rdc::StartJobStatsResponse* reply) override;
+
+  ::grpc::Status GetJobStats(::grpc::ServerContext* context,
+                             const ::rdc::GetJobStatsRequest* request,
+                             ::rdc::GetJobStatsResponse* reply) override;
+
+  ::grpc::Status StopJobStats(::grpc::ServerContext* context,
+                              const ::rdc::StopJobStatsRequest* request,
+                              ::rdc::StopJobStatsResponse* reply) override;
+
+  ::grpc::Status RemoveJob(::grpc::ServerContext* context, const ::rdc::RemoveJobRequest* request,
+                           ::rdc::RemoveJobResponse* reply) override;
+
+  ::grpc::Status RemoveAllJob(::grpc::ServerContext* context, const ::rdc::Empty* request,
+                              ::rdc::RemoveAllJobResponse* reply) override;
+
+  ::grpc::Status DiagnosticRun(
+      ::grpc::ServerContext* context, const ::rdc::DiagnosticRunRequest* request,
+      ::grpc::ServerWriter< ::rdc::DiagnosticRunResponse>* writer) override;
+
+  ::grpc::Status DiagnosticTestCaseRun(
+      ::grpc::ServerContext* context, const ::rdc::DiagnosticTestCaseRunRequest* request,
+      ::grpc::ServerWriter< ::rdc::DiagnosticTestCaseRunResponse>* writer) override;
+
+  ::grpc::Status GetMixedComponentVersion(::grpc::ServerContext* context,
+                                          const ::rdc::GetMixedComponentVersionRequest* request,
+                                          ::rdc::GetMixedComponentVersionResponse* reply) override;
+
+  ::grpc::Status SetPolicy(::grpc::ServerContext* context, const ::rdc::SetPolicyRequest* request,
+                           ::rdc::SetPolicyResponse* reply) override;
+
+  ::grpc::Status GetPolicy(::grpc::ServerContext* context, const ::rdc::GetPolicyRequest* request,
+                           ::rdc::GetPolicyResponse* reply) override;
+
+  ::grpc::Status DeletePolicy(::grpc::ServerContext* context,
+                              const ::rdc::DeletePolicyRequest* request,
+                              ::rdc::DeletePolicyResponse* reply) override;
+
+  ::grpc::Status RegisterPolicy(
+      ::grpc::ServerContext* context, const ::rdc::RegisterPolicyRequest* request,
+      ::grpc::ServerWriter< ::rdc::RegisterPolicyResponse>* stream) override;
+
+  ::grpc::Status UnRegisterPolicy(::grpc::ServerContext* context,
+                                  const ::rdc::UnRegisterPolicyRequest* request,
+                                  ::rdc::UnRegisterPolicyResponse* reply) override;
+
+  ::grpc::Status GetTopology(::grpc::ServerContext* context,
+                             const ::rdc::GetTopologyRequest* request,
+                             ::rdc::GetTopologyResponse* reply) override;
+
+  ::grpc::Status GetLinkStatus(::grpc::ServerContext* context, const ::rdc::Empty* request,
+                               ::rdc::GetLinkStatusResponse* reply) override;
+
+  ::grpc::Status SetHealth(::grpc::ServerContext* context, const ::rdc::SetHealthRequest* request,
+                           ::rdc::SetHealthResponse* reply) override;
+
+  ::grpc::Status GetHealth(::grpc::ServerContext* context, const ::rdc::GetHealthRequest* request,
+                           ::rdc::GetHealthResponse* reply) override;
+
+  ::grpc::Status CheckHealth(::grpc::ServerContext* context,
+                             const ::rdc::CheckHealthRequest* request,
+                             ::rdc::CheckHealthResponse* reply) override;
+
+  ::grpc::Status ClearHealth(::grpc::ServerContext* context,
+                             const ::rdc::ClearHealthRequest* request,
+                             ::rdc::ClearHealthResponse* reply) override;
+
+  ::grpc::Status SetConfig(::grpc::ServerContext* context, const ::rdc::SetConfigRequest* request,
+                           ::rdc::SetConfigResponse* reply) override;
+
+  ::grpc::Status GetConfig(::grpc::ServerContext* context, const ::rdc::GetConfigRequest* request,
+                           ::rdc::GetConfigResponse* reply) override;
+
+  ::grpc::Status ClearConfig(::grpc::ServerContext* context,
+                             const ::rdc::ClearConfigRequest* request,
+                             ::rdc::ClearConfigResponse* reply) override;
+
+  ::grpc::Status GetNumPartition(::grpc::ServerContext* context,
+                                 const ::rdc::GetNumPartitionRequest* request,
+                                 ::rdc::GetNumPartitionResponse* reply) override;
+
+  ::grpc::Status GetInstanceProfile(::grpc::ServerContext* context,
+                                    const ::rdc::GetInstanceProfileRequest* request,
+                                    ::rdc::GetInstanceProfileResponse* reply) override;
+
+ private:
+  bool copy_gpu_usage_info(const rdc_gpu_usage_info_t& src, ::rdc::GpuUsageInfo* target);
+  rdc_handle_t rdc_handle_;
+
+  struct policy_thread_context {
+    pthread_mutex_t mutex;
+    pthread_cond_t cond;
+    rdc_policy_callback_response_t response;
+    bool start;
+  };
+  // map for group_id and thread context
+  static std::map<uint32_t, struct policy_thread_context*> policy_threads_;
+  static int PolicyCallback(rdc_policy_callback_response_t* userData);
+};
+
+}  // namespace rdc
+}  // namespace amd
+
+#endif  // SERVER_INCLUDE_RDC_RDC_API_SERVICE_H_
@@ -0,0 +1,87 @@
+/*
+Copyright (c) 2019 - present Advanced Micro Devices, Inc. All rights reserved.
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in
+all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
+THE SOFTWARE.
+*/
+#ifndef SERVER_INCLUDE_RDC_RDC_SERVER_MAIN_H_
+#define SERVER_INCLUDE_RDC_RDC_SERVER_MAIN_H_
+
+#include <grpcpp/grpcpp.h>
+
+#include <memory>
+#include <string>
+
+#include "rdc/rdc_admin_service.h"
+#include "rdc/rdc_api_service.h"
+
+#define RDC_SERVER_VERSION_MAJOR 1
+#define RDC_SERVER_VERSION_MINOR 1
+#define RDC_SERVER_VERSION_RELEASE 0
+
+#define RDC_SERVER_VERSION_CREATE_STRING(MAJOR, MINOR, RELEASE) (#MAJOR "." #MINOR "." #RELEASE)
+#define RDC_SERVER_VERSION_EXPAND_PARTS(MAJOR_STR, MINOR_STR, RELEASE_STR) RDC_SERVER_VERSION_CREATE_STRING(MAJOR_STR, MINOR_STR, RELEASE_STR)
+#define RDC_SERVER_VERSION_STRING RDC_SERVER_VERSION_EXPAND_PARTS(RDC_SERVER_VERSION_MAJOR, RDC_SERVER_VERSION_MINOR, RDC_SERVER_VERSION_RELEASE)
+
+#define Q(x) #x
+#define QUOTE(x) Q(x)
+
+typedef struct {
+  std::string listen_address;
+  std::string listen_port;
+  bool no_authentication;
+  bool use_pinned_certs;
+  bool log_dbg;
+} RdcdCmdLineOpts;
+
+class RDCServer {
+ public:
+  RDCServer();
+  ~RDCServer();
+
+  void Initialize(RdcdCmdLineOpts* cl);
+
+  void Run(void);
+  void ShutDown(void);
+
+  bool start_rdc_admin_service(void) const { return start_rdc_admin_service_; }
+  void set_start_rdc_admin_service(bool s) { start_rdc_admin_service_ = s; }
+
+  bool start_api_service(void) const { return start_api_service_; }
+  void set_start_api_service(bool s) { start_api_service_ = s; }
+
+  bool secure_creds(void) const { return secure_creds_; }
+  void set_secure_creds(bool s) { secure_creds_ = s; }
+
+ private:
+  void HandleSignal(int sig);
+  std::string server_address_;
+  std::unique_ptr<::grpc::Server> server_;
+  bool secure_creds_;
+  bool use_pinned_certs_;
+  bool log_debug_;
+  RdcdCmdLineOpts* cmd_line_;
+
+  bool start_rdc_admin_service_;
+  amd::rdc::RDCAdminServiceImpl* rdc_admin_service_;
+
+  bool start_api_service_;
+  amd::rdc::RdcAPIServiceImpl* api_service_;
+};
+
+#endif  // SERVER_INCLUDE_RDC_RDC_SERVER_MAIN_H_