1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
|
// unit-test-module-ptr.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(modulePtr)
{
const char* testModuleSource = R"(
module test_module;
public void atomicFunc(__ref Atomic<int> ptr) {
ptr.add(1);
}
)";
const char* testSource = R"(
import "test_module";
RWStructuredBuffer<Atomic<int>> input0;
[shader("compute")]
[numthreads(1,1,1)]
void computeMain(uint3 workGroup : SV_GroupID)
{
atomicFunc(input0[0]);
}
)";
ComPtr<ISlangMutableFileSystem> memoryFileSystem =
ComPtr<ISlangMutableFileSystem>(new Slang::MemoryFileSystem());
ComPtr<slang::IGlobalSession> globalSession;
SLANG_CHECK(slang_createGlobalSession(SLANG_API_VERSION, globalSession.writeRef()) == SLANG_OK);
slang::TargetDesc targetDesc = {};
targetDesc.format = SLANG_SPIRV;
targetDesc.profile = globalSession->findProfile("spirv_1_5");
slang::SessionDesc sessionDesc = {};
sessionDesc.targetCount = 1;
sessionDesc.targets = &targetDesc;
sessionDesc.compilerOptionEntryCount = 0;
sessionDesc.fileSystem = memoryFileSystem;
// Precompile test_module to file.
{
ComPtr<slang::ISession> session;
SLANG_CHECK(globalSession->createSession(sessionDesc, session.writeRef()) == SLANG_OK);
ComPtr<slang::IBlob> diagnosticBlob;
auto module = session->loadModuleFromSourceString(
"test_module",
"test_module.slang",
testModuleSource,
diagnosticBlob.writeRef());
SLANG_CHECK(module != nullptr);
ComPtr<slang::IBlob> moduleBlob;
module->serialize(moduleBlob.writeRef());
memoryFileSystem->saveFile(
"test_module.slang-module",
moduleBlob->getBufferPointer(),
moduleBlob->getBufferSize());
}
// compile test.
{
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());
ComPtr<slang::IBlob> code;
linkedProgram->getTargetCode(0, code.writeRef(), diagnosticBlob.writeRef());
SLANG_CHECK(code->getBufferSize() > 0);
}
}
|
