blob: f3df15b64ae44fae7d30b3c1c8a2d4e04191406b (
plain)
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
|
// unit-test-translation-unit-import.cpp
#include "../../source/core/slang-io.h"
#include "../../source/core/slang-process.h"
#include "slang-com-ptr.h"
#include "slang.h"
#include "tools/unit-test/slang-unit-test.h"
#include <stdio.h>
#include <stdlib.h>
using namespace Slang;
// Test that the IComponentType::getTargetCode API supports
// compiling a program with multiple entrypoints and retrieving a single
// compiled module that contains all the entrypoints.
//
SLANG_UNIT_TEST(getTargetCode)
{
// Source for a module that contains an undecorated entrypoint.
const char* userSourceBody = R"(
[shader("fragment")]
float4 fragMain(float4 pos:SV_Position) : SV_Target
{
return pos;
}
[shader("vertex")]
float4 vertMain(float4 pos) : SV_Position
{
return pos;
}
)";
String userSource = userSourceBody;
ComPtr<slang::IGlobalSession> globalSession;
SLANG_CHECK(slang_createGlobalSession(SLANG_API_VERSION, globalSession.writeRef()) == SLANG_OK);
slang::TargetDesc targetDesc = {};
// Request SPIR-V disassembly so we can check the content.
targetDesc.format = SLANG_SPIRV_ASM;
targetDesc.profile = globalSession->findProfile("sm_5_0");
slang::SessionDesc sessionDesc = {};
sessionDesc.targetCount = 1;
sessionDesc.targets = &targetDesc;
ComPtr<slang::ISession> session;
SLANG_CHECK(globalSession->createSession(sessionDesc, session.writeRef()) == SLANG_OK);
ComPtr<slang::IBlob> diagnosticBlob;
auto module = session->loadModuleFromSourceString(
"m",
"m.slang",
userSourceBody,
diagnosticBlob.writeRef());
SLANG_CHECK(module != nullptr);
ComPtr<slang::IComponentType> linkedProgram;
module->link(linkedProgram.writeRef(), diagnosticBlob.writeRef());
SLANG_CHECK(linkedProgram != nullptr);
ComPtr<slang::IBlob> code;
linkedProgram->getTargetCode(0, code.writeRef(), diagnosticBlob.writeRef());
SLANG_CHECK(code != nullptr);
SLANG_CHECK(code->getBufferSize() != 0);
UnownedStringSlice resultStr = UnownedStringSlice((char*)code->getBufferPointer());
// Make sure the spirv disassembly contains both entrypoint names.
SLANG_CHECK(resultStr.indexOf(toSlice("fragMain")) != -1);
SLANG_CHECK(resultStr.indexOf(toSlice("vertMain")) != -1);
}
|
