summaryrefslogtreecommitdiffstats
path: root/tools/slang-unit-test
diff options
context:
space:
mode:
authorYong He <yonghe@outlook.com>2025-04-28 11:42:22 -0700
committerGitHub <noreply@github.com>2025-04-28 11:42:22 -0700
commitc39c29bf4c52a85d7c83cc8b66ae45e265f9e078 (patch)
tree969339828d49d7db92ed9294a17bd34cc021db84 /tools/slang-unit-test
parent8f6c6e333c06ae1c3b9f00396563c14a2ae09b4d (diff)
Add Slang Byte Code generation and interpreter. (#6896)
* Add Slang Byte Code generation and interpreter. * Fix compile issues. * format code * More compile fix. * Fix clang issue. * Fix more clang issues. * Another clang fix. * Fix clang issues. * Fix another clang issue. * Fix wasm build. * Update building.md * Fix test-server. * Fix compile error. * Fix bug. --------- Co-authored-by: slangbot <186143334+slangbot@users.noreply.github.com>
Diffstat (limited to 'tools/slang-unit-test')
-rw-r--r--tools/slang-unit-test/unit-test-slang-vm.cpp102
1 files changed, 102 insertions, 0 deletions
diff --git a/tools/slang-unit-test/unit-test-slang-vm.cpp b/tools/slang-unit-test/unit-test-slang-vm.cpp
new file mode 100644
index 000000000..0f5e0f9f3
--- /dev/null
+++ b/tools/slang-unit-test/unit-test-slang-vm.cpp
@@ -0,0 +1,102 @@
+// unit-test-slang-vm.cpp
+
+#include "core/slang-memory-file-system.h"
+#include "slang-com-ptr.h"
+#include "slang.h"
+#include "unit-test/slang-unit-test.h"
+
+#include <stdio.h>
+#include <stdlib.h>
+
+using namespace Slang;
+
+SLANG_UNIT_TEST(slangVM)
+{
+ const char* testSource = R"(
+ int one() { return 1; }
+ int sum(int x)
+ {
+ int result = 0;
+ for (int i = 0; i <= x; i++)
+ {
+ result += i;
+ }
+ return result + one();
+ }
+ [shader("dispatch")]
+ int dispatchMain(uniform int2 v, out int c)
+ {
+ int a = v.x;
+ int b = v.y;
+ int tmp = 0;
+ if (a > 0)
+ tmp = a + b;
+ else
+ tmp = b - a;
+ tmp += sum(b);
+ c = tmp;
+ return 100;
+ }
+ )";
+
+ // Create Slang session and compile code.
+ ComPtr<slang::IBlob> code;
+ String disasmText;
+ {
+ ComPtr<slang::IGlobalSession> globalSession;
+ SLANG_CHECK(
+ slang_createGlobalSession(SLANG_API_VERSION, globalSession.writeRef()) == SLANG_OK);
+ slang::TargetDesc targetDesc = {};
+ targetDesc.format = SLANG_HOST_VM;
+ slang::SessionDesc sessionDesc = {};
+ sessionDesc.targetCount = 1;
+ sessionDesc.targets = &targetDesc;
+ sessionDesc.compilerOptionEntryCount = 0;
+
+ ComPtr<slang::ISession> session;
+ SLANG_CHECK(globalSession->createSession(sessionDesc, session.writeRef()) == SLANG_OK);
+
+ ComPtr<slang::IBlob> diagnosticBlob;
+ auto module = session->loadModuleFromSourceString(
+ "test",
+ "test.slang",
+ testSource,
+ diagnosticBlob.writeRef());
+ SLANG_CHECK(module != nullptr);
+
+ ComPtr<slang::IComponentType> linkedProgram;
+ module->link(linkedProgram.writeRef());
+
+
+ linkedProgram->getTargetCode(0, code.writeRef(), diagnosticBlob.writeRef());
+
+ SLANG_CHECK(code->getBufferSize() > 0);
+
+ ComPtr<slang::IBlob> disasmBlob;
+ SLANG_CHECK(slang_disassembleByteCode(code, disasmBlob.writeRef()) == SLANG_OK);
+ disasmText = (const char*)disasmBlob->getBufferPointer();
+ SLANG_CHECK(disasmText.indexOf("ret") != -1);
+ }
+
+ // Create a byte code runner and interpret the code.
+ ComPtr<slang::IByteCodeRunner> runner;
+ slang::ByteCodeRunnerDesc runnerDesc = {};
+ SLANG_CHECK(slang_createByteCodeRunner(&runnerDesc, runner.writeRef()) == SLANG_OK);
+ SLANG_CHECK(runner->loadModule(code) == SLANG_OK);
+ SLANG_CHECK(runner->selectFunctionByIndex(0) == SLANG_OK);
+ struct Params
+ {
+ int a;
+ int b;
+ int* result;
+ };
+ int result = 0;
+ Params params = {1, 2, &result};
+ SLANG_CHECK(runner->execute(&params, sizeof(params)) == SLANG_OK);
+ SLANG_CHECK(result == 7);
+
+ size_t returnValSize = 0;
+ int* returnVal = (int*)runner->getReturnValue(&returnValSize);
+ SLANG_CHECK(returnValSize == sizeof(int));
+ SLANG_CHECK(*returnVal == 100);
+}