From 6b35fc4cf6e3cf08e5c24c3d0cf6f83b088feb9e Mon Sep 17 00:00:00 2001
From: Graham Sider <Graham.Sider@amd.com>
Date: Thu, 30 Sep 2021 18:04:51 -0400
Subject: [PATCH] kfdtest: Remove IsaGen from KFDLocalMemoryTest

Replace with LLVM-based Assembler.

Signed-off-by: Graham Sider <Graham.Sider@amd.com>
Change-Id: Ie83d27b6a93ac1b5169a830a7f274e360a31023e


[ROCm/ROCR-Runtime commit: 549f7cdce20156e4ef78ac5b44dae662e21a4a25]
---
 .../tests/kfdtest/src/KFDLocalMemoryTest.cpp        | 13 ++++---------
 .../tests/kfdtest/src/KFDLocalMemoryTest.hpp        |  6 +-----
 2 files changed, 5 insertions(+), 14 deletions(-)

diff --git a/projects/rocr-runtime/tests/kfdtest/src/KFDLocalMemoryTest.cpp b/projects/rocr-runtime/tests/kfdtest/src/KFDLocalMemoryTest.cpp
index a27b502f97..b37528c651 100644
--- a/projects/rocr-runtime/tests/kfdtest/src/KFDLocalMemoryTest.cpp
+++ b/projects/rocr-runtime/tests/kfdtest/src/KFDLocalMemoryTest.cpp
@@ -33,18 +33,12 @@ void KFDLocalMemoryTest::SetUp() {
 
     KFDBaseComponentTest::SetUp();
 
-    m_pIsaGen = IsaGenerator::Create(m_FamilyId);
-
     ROUTINE_END
 }
 
 void KFDLocalMemoryTest::TearDown() {
     ROUTINE_START
 
-    if (m_pIsaGen)
-        delete m_pIsaGen;
-    m_pIsaGen = NULL;
-
     KFDBaseComponentTest::TearDown();
 
     ROUTINE_END
@@ -107,7 +101,7 @@ TEST_F(KFDLocalMemoryTest, BasicTest) {
 
     srcSysBuffer.Fill(0x01010101);
 
-    m_pIsaGen->GetCopyDwordIsa(isaBuffer);
+    ASSERT_SUCCESS(m_pAsm->RunAssembleBuf(CopyDwordIsa, isaBuffer.As<char*>()));
 
     ASSERT_SUCCESS(hsaKmtMapMemoryToGPUNodes(srcLocalBuffer.As<void*>(), srcLocalBuffer.Size(), &AlternateVAGPU,
                         mapFlags, 1, reinterpret_cast<HSAuint32 *>(&defaultGPUNode)));
@@ -164,7 +158,7 @@ TEST_F(KFDLocalMemoryTest, VerifyContentsAfterUnmapAndMap) {
 
     SysBufferA.Fill(0x01010101);
 
-    m_pIsaGen->GetCopyDwordIsa(isaBuffer);
+    ASSERT_SUCCESS(m_pAsm->RunAssembleBuf(CopyDwordIsa, isaBuffer.As<char*>()));
 
     ASSERT_SUCCESS(queue.Create(defaultGPUNode));
     queue.SetSkipWaitConsump(0);
@@ -303,7 +297,8 @@ TEST_F(KFDLocalMemoryTest, Fragmentation) {
     PM4Queue queue;
     ASSERT_SUCCESS(queue.Create(defaultGPUNode));
     HsaMemoryBuffer isaBuffer(PAGE_SIZE, defaultGPUNode);
-    m_pIsaGen->GetCopyDwordIsa(isaBuffer);
+
+    ASSERT_SUCCESS(m_pAsm->RunAssembleBuf(CopyDwordIsa, isaBuffer.As<char*>()));
 
     /* Allocate and test memory using the strategy explained at the top */
     HSAKMT_STATUS status;
diff --git a/projects/rocr-runtime/tests/kfdtest/src/KFDLocalMemoryTest.hpp b/projects/rocr-runtime/tests/kfdtest/src/KFDLocalMemoryTest.hpp
index 519081cfc0..b3a9b2add6 100644
--- a/projects/rocr-runtime/tests/kfdtest/src/KFDLocalMemoryTest.hpp
+++ b/projects/rocr-runtime/tests/kfdtest/src/KFDLocalMemoryTest.hpp
@@ -26,20 +26,16 @@
 
 #include <gtest/gtest.h>
 
-#include "IsaGenerator.hpp"
 #include "KFDBaseComponentTest.hpp"
 
 class KFDLocalMemoryTest : public KFDBaseComponentTest {
  public:
-    KFDLocalMemoryTest() :m_pIsaGen(NULL) {}
+    KFDLocalMemoryTest() {}
     ~KFDLocalMemoryTest() {}
 
  protected:
     virtual void SetUp();
     virtual void TearDown();
-
- protected:  // Members
-    IsaGenerator* m_pIsaGen;
 };
 
 #endif  // __KFD_LOCALMEMORY_TEST__H__