Page migration reporting (#651)

* Page migration reporting support

* Page migration: Update parser and reporting

Container does not lave latest KFD header, so CI might fail

* Add kfd_ioctl.h

* Formatting

* Update get_key

- get key was not used (and shouldn't be), so delete it

* clang-tidy fixes

* Tests for page migration

* Apply suggestions from code review

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>

* Update tests/bin/page-migration/CMakeLists.txt

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>

* Update page-migration test app

- add hipHostRegister to register mmap'ed allocation with HIP
- misc cleanup and reorg
- remove HSA_XNACK=1 from test env

* Update lib/rocprofiler-sdk/tests/page_migration.cpp

- fix compilation error

* Minor updates (reorg, rename)

* Page migration reporting support

* Page migration: Update parser and reporting

Container does not lave latest KFD header, so CI might fail

* Update page migration tests, fix trigger types

* Page Migration Tracing Support Refactoring (#753)

* Reorganization

* Update page migration init/fini

* Formatting

* Update page_migration.cpp

- change logging severity

* Skip test if KFD does not support page migration reporting

* Rework skipping test if KFD does not support page migration

* Fix event trigger enum values

* Fix clang-diagnostic-unused-const-variable

---------

Co-authored-by: github-actions[bot] <41898282+github-actions[bot]@users.noreply.github.com>
Co-authored-by: Jonathan R. Madsen <jonathanrmadsen@gmail.com>
Co-authored-by: Jonathan R. Madsen <jrmadsen@users.noreply.github.com>

tests/common/defines.hpp

@@ -33,8 +33,8 @@
                       << " failed with error code " << status_name << " (" << CHECKSTATUS          \
                       << "): " << status_msg << std::endl;                                         \
             std::stringstream errmsg{};                                                            \
-            errmsg << "[" #result "][" << __FILE__ << ":" << __LINE__ << "] " << msg " failure ("  \
-                   << status_name << ": " << status_msg << ")";                                    \
+            errmsg << "[" #result "][" << __FILE__ << ":" << __LINE__ << "] " << msg               \
+                   << " failure (" << status_name << ": " << status_msg << ")";                    \
             throw std::runtime_error(errmsg.str());                                                \
         }                                                                                          \
     }

tests/common/serialization.hpp

@@ -391,6 +391,90 @@ save(ArchiveT& ar, rocprofiler_buffer_tracing_memory_copy_record_t data)
     SAVE_DATA_FIELD(src_agent_id);
 }
 
+template <typename ArchiveT>
+void
+save(ArchiveT& ar, const rocprofiler_buffer_tracing_page_migration_record_t& data)
+{
+    SAVE_DATA_FIELD(size);
+    SAVE_DATA_FIELD(kind);
+    SAVE_DATA_FIELD(operation);
+    SAVE_DATA_FIELD(start_timestamp);
+    SAVE_DATA_FIELD(end_timestamp);
+    SAVE_DATA_FIELD(pid);
+
+    switch(data.operation)
+    {
+        case ROCPROFILER_PAGE_MIGRATION_PAGE_FAULT:
+        {
+            ar(make_nvp("page_fault", data.page_fault));
+            break;
+        }
+        case ROCPROFILER_PAGE_MIGRATION_PAGE_MIGRATE:
+        {
+            ar(make_nvp("page_migrate", data.page_migrate));
+            break;
+        }
+        case ROCPROFILER_PAGE_MIGRATION_QUEUE_SUSPEND:
+        {
+            ar(make_nvp("queue_suspend", data.queue_suspend));
+            break;
+        }
+        case ROCPROFILER_PAGE_MIGRATION_UNMAP_FROM_GPU:
+        {
+            ar(make_nvp("unmap_from_gpu", data.unmap_from_gpu));
+            break;
+        }
+        case ROCPROFILER_PAGE_MIGRATION_NONE:
+        case ROCPROFILER_PAGE_MIGRATION_LAST:
+        {
+            throw std::runtime_error{"unsupported page migration operation type"};
+            break;
+        }
+    }
+}
+
+template <typename ArchiveT>
+void
+save(ArchiveT& ar, const rocprofiler_buffer_tracing_page_migration_page_fault_record_t& data)
+{
+    SAVE_DATA_FIELD(node_id);
+    SAVE_DATA_FIELD(address);
+    SAVE_DATA_FIELD(read_fault);
+    SAVE_DATA_FIELD(migrated);
+}
+
+template <typename ArchiveT>
+void
+save(ArchiveT& ar, const rocprofiler_buffer_tracing_page_migration_page_migrate_record_t& data)
+{
+    SAVE_DATA_FIELD(start_addr);
+    SAVE_DATA_FIELD(end_addr);
+    SAVE_DATA_FIELD(from_node);
+    SAVE_DATA_FIELD(to_node);
+    SAVE_DATA_FIELD(prefetch_node);
+    SAVE_DATA_FIELD(preferred_node);
+    SAVE_DATA_FIELD(trigger);
+}
+
+template <typename ArchiveT>
+void
+save(ArchiveT& ar, const rocprofiler_buffer_tracing_page_migration_queue_suspend_record_t& data)
+{
+    SAVE_DATA_FIELD(node_id);
+    SAVE_DATA_FIELD(trigger);
+    SAVE_DATA_FIELD(rescheduled);
+}
+
+template <typename ArchiveT>
+void
+save(ArchiveT& ar, const rocprofiler_buffer_tracing_page_migration_unmap_from_gpu_record_t& data)
+{
+    SAVE_DATA_FIELD(node_id);
+    SAVE_DATA_FIELD(start_addr);
+    SAVE_DATA_FIELD(end_addr);
+    SAVE_DATA_FIELD(trigger);
+}
+
 template <typename ArchiveT>
 void
 save(ArchiveT& ar, rocprofiler_buffer_tracing_scratch_memory_record_t data)