Add initial support for getting process information

Added implementation of and tests for rsmi_dev_compute_process_info_by_pid_get() and rsmi_dev_compute_process_info_get() Change-Id: I4c4f5f39fe6701da37916c9ad41449b5d35ac7af
2019-07-03 11:29:19 -05:00
commit 9b93cbe21d
@@ -94,7 +94,6 @@ void TestMemUtilRead::Run(void) {
  rsmi_status_t err;
  uint64_t total;
  uint64_t usage;
-  uint32_t mem_busy_percent;

  TestBase::Run();

@@ -0,0 +1,181 @@
+/*
+ * =============================================================================
+ *   ROC Runtime Conformance Release License
+ * =============================================================================
+ * The University of Illinois/NCSA
+ * Open Source License (NCSA)
+ *
+ * Copyright (c) 2019, Advanced Micro Devices, Inc.
+ * All rights reserved.
+ *
+ * Developed by:
+ *
+ *                 AMD Research and AMD ROC Software Development
+ *
+ *                 Advanced Micro Devices, Inc.
+ *
+ *                 www.amd.com
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to
+ * deal with 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:
+ *
+ *  - Redistributions of source code must retain the above copyright notice,
+ *    this list of conditions and the following disclaimers.
+ *  - Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimers in
+ *    the documentation and/or other materials provided with the distribution.
+ *  - Neither the names of <Name of Development Group, Name of Institution>,
+ *    nor the names of its contributors may be used to endorse or promote
+ *    products derived from this Software without specific prior written
+ *    permission.
+ *
+ * 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 CONTRIBUTORS 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 WITH THE SOFTWARE.
+ *
+ */
+
+#include <stdint.h>
+#include <stddef.h>
+
+#include <iostream>
+#include <string>
+
+#include "gtest/gtest.h"
+#include "rocm_smi/rocm_smi.h"
+#include "rocm_smi_test/functional/process_info_read.h"
+#include "rocm_smi_test/test_common.h"
+
+TestProcInfoRead::TestProcInfoRead() : TestBase() {
+  set_title("RSMI Process Info Read Test");
+  set_description("This test verifies that process information such as the "
+                             "process ID, PASID, etc. can be read properly.");
+}
+
+TestProcInfoRead::~TestProcInfoRead(void) {
+}
+
+void TestProcInfoRead::SetUp(void) {
+  TestBase::SetUp();
+
+  return;
+}
+
+void TestProcInfoRead::DisplayTestInfo(void) {
+  TestBase::DisplayTestInfo();
+}
+
+void TestProcInfoRead::DisplayResults(void) const {
+  TestBase::DisplayResults();
+  return;
+}
+
+void TestProcInfoRead::Close() {
+  // This will close handles opened within rsmitst utility calls and call
+  // rsmi_shut_down(), so it should be done after other hsa cleanup
+  TestBase::Close();
+}
+
+static void dumpProcess(rsmi_process_info_t *p) {
+  assert(p != nullptr);
+  std::cout << "ProcessID: " << p->process_id << " ";
+  std::cout << "PASID: " << p->pasid << " ";
+  std::cout << std::endl;
+}
+void TestProcInfoRead::Run(void) {
+  rsmi_status_t err;
+  uint32_t num_proc_found;
+  uint32_t val_ui32;
+  rsmi_process_info_t *procs = nullptr;
+
+  TestBase::Run();
+
+  err = rsmi_compute_process_info_get(nullptr, &num_proc_found);
+  if (err != RSMI_STATUS_SUCCESS) {
+    if (err == RSMI_STATUS_FILE_ERROR) {
+      IF_VERB(STANDARD) {
+        std::cout << "\t**Process info. read: Not supported on this machine"
+                                                                 << std::endl;
+        return;
+      }
+    } else {
+      CHK_ERR_ASRT(err)
+    }
+  } else {
+    IF_VERB(STANDARD) {
+      std::cout << "\t** "  << std::dec << num_proc_found <<
+                                          " GPU processes found" << std::endl;
+    }
+  }
+
+  if (num_proc_found == 0) {
+    return;
+  }
+  procs = new rsmi_process_info_t[num_proc_found];
+
+  val_ui32 = num_proc_found;
+  err = rsmi_compute_process_info_get(procs, &val_ui32);
+  if (err != RSMI_STATUS_SUCCESS) {
+    if (err == RSMI_STATUS_INSUFFICIENT_SIZE) {
+      IF_VERB(STANDARD) {
+        std::cout << "\t** " << val_ui32 <<
+         " processes were read, but more became available that were unread."
+                                                                 << std::endl;
+        for (uint32_t i = 0; i < val_ui32; ++i) {
+          dumpProcess(&procs[i]);
+        }
+
+        return;
+      }
+    } else {
+      CHK_ERR_ASRT(err)
+    }
+  } else {
+    IF_VERB(STANDARD) {
+      std::cout << "\t** Processes currently using GPU: " << std::endl;
+      for (uint32_t i = 0; i < val_ui32; ++i) {
+        dumpProcess(&procs[i]);
+      }
+    }
+  }
+
+  // Reset to the number we actually read
+  num_proc_found = val_ui32;
+  if (num_proc_found) {
+    rsmi_process_info_t proc_info;
+
+    err = rsmi_compute_process_info_by_pid_get(procs[0].process_id,
+                                                                  &proc_info);
+    if (err == RSMI_STATUS_NOT_FOUND) {
+      std::cout <<
+       "\t** WARNING: rsmi_compute_process_info_get() found process " <<
+         procs[0].process_id << ", but subsequently, "
+                       "rsmi_compute_process_info_by_pid_get() did not"
+                                      " find this same process." << std::endl;
+    } else {
+      CHK_ERR_ASRT(err)
+      ASSERT_EQ(proc_info.process_id, procs[0].process_id);
+      ASSERT_EQ(proc_info.pasid, procs[0].pasid);
+    }
+  }
+  if (num_proc_found > 1) {
+    rsmi_process_info_t tmp_proc;
+    val_ui32 = 1;
+    err = rsmi_compute_process_info_get(&tmp_proc, &val_ui32);
+
+    if (err != RSMI_STATUS_INSUFFICIENT_SIZE) {
+      std::cout << "Expected rsmi_compute_process_info_get() to tell us"
+        " there are more processes available, but instead go return code " <<
+                                                              err << std::endl;
+    }
+  }
+  delete []procs;
+}
@@ -0,0 +1,73 @@
+/*
+ * =============================================================================
+ *   ROC Runtime Conformance Release License
+ * =============================================================================
+ * The University of Illinois/NCSA
+ * Open Source License (NCSA)
+ *
+ * Copyright (c) 2019, Advanced Micro Devices, Inc.
+ * All rights reserved.
+ *
+ * Developed by:
+ *
+ *                 AMD Research and AMD ROC Software Development
+ *
+ *                 Advanced Micro Devices, Inc.
+ *
+ *                 www.amd.com
+ *
+ * Permission is hereby granted, free of charge, to any person obtaining a copy
+ * of this software and associated documentation files (the "Software"), to
+ * deal with 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:
+ *
+ *  - Redistributions of source code must retain the above copyright notice,
+ *    this list of conditions and the following disclaimers.
+ *  - Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimers in
+ *    the documentation and/or other materials provided with the distribution.
+ *  - Neither the names of <Name of Development Group, Name of Institution>,
+ *    nor the names of its contributors may be used to endorse or promote
+ *    products derived from this Software without specific prior written
+ *    permission.
+ *
+ * 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 CONTRIBUTORS 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 WITH THE SOFTWARE.
+ *
+ */
+#ifndef TESTS_ROCM_SMI_TEST_FUNCTIONAL_PROCESS_INFO_READ_H_
+#define TESTS_ROCM_SMI_TEST_FUNCTIONAL_PROCESS_INFO_READ_H_
+
+#include "rocm_smi_test/test_base.h"
+
+class TestProcInfoRead : public TestBase {
+ public:
+    TestProcInfoRead();
+
+  // @Brief: Destructor for test case of TestProcInfoRead
+  virtual ~TestProcInfoRead();
+
+  // @Brief: Setup the environment for measurement
+  virtual void SetUp();
+
+  // @Brief: Core measurement execution
+  virtual void Run();
+
+  // @Brief: Clean up and retrive the resource
+  virtual void Close();
+
+  // @Brief: Display  results
+  virtual void DisplayResults() const;
+
+  // @Brief: Display information about what this test does
+  virtual void DisplayTestInfo(void);
+};
+
+#endif  // TESTS_ROCM_SMI_TEST_FUNCTIONAL_PROCESS_INFO_READ_H_
@@ -72,7 +72,8 @@
 #include "functional/err_cnt_read.h"
 #include "functional/mem_util_read.h"
 #include "functional/id_info_read.h"
-#include "rocm_smi_test/functional/perf_cntr_read_write.h"
+#include "functional/perf_cntr_read_write.h"
+#include "functional/process_info_read.h"

 static RSMITstGlobals *sRSMIGlvalues = nullptr;

@@ -199,10 +200,15 @@ TEST(rsmitstReadOnly, TestIdInfoRead) {
  TestIdInfoRead tst;
  RunGenericTest(&tst);
 }
-TEST(rsmitstreadWrite, TestPerfCntrReadWrite) {
+TEST(rsmitstReadWrite, TestPerfCntrReadWrite) {
  TestPerfCntrReadWrite tst;
  RunGenericTest(&tst);
 }
+TEST(rsmitstReadOnly, TestProcInfoRead) {
+  TestProcInfoRead tst;
+  RunGenericTest(&tst);
+}
+
 int main(int argc, char** argv) {
  ::testing::InitGoogleTest(&argc, argv);