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
|
/*========================== begin_copyright_notice ============================
Copyright (C) 2023 Intel Corporation
SPDX-License-Identifier: MIT
============================= end_copyright_notice ===========================*/
// This file contains interface for compilation of a SPIR-V module that
// consists of combined SPMD and ESIMD parts.
#include <string>
#include <llvm/ADT/ArrayRef.h>
#include "TranslationBlock.h"
#pragma once
// Forward declarations.
class ShaderHash;
namespace IGC {
class CPlatform;
}
namespace IGC {
namespace VLD {
enum class SPIRVTypeEnum { SPIRV_SPMD, SPIRV_ESIMD, SPIRV_SPMD_AND_ESIMD, SPIRV_TYPE_UNKNOWN };
struct SPVMetadata {
SPIRVTypeEnum SpirvType = SPIRVTypeEnum::SPIRV_TYPE_UNKNOWN;
std::vector<std::string> ExportedFunctions;
std::vector<std::string> ImportedFunctions;
bool HasEntryPoints = false;
int ForcedSubgroupSize = 0;
};
using SPVTranslationPair = std::pair<SPVMetadata, TC::STB_TranslateInputArgs>;
// This function detects if binary passed in pInputArgs is SPMD, ESIMD or
// SPMD+ESIMD
bool TranslateBuildSPMDAndESIMD(const TC::STB_TranslateInputArgs *pInputArgs, TC::STB_TranslateOutputArgs *pOutputArgs,
TC::TB_DATA_FORMAT inputDataFormatTemp, const IGC::CPlatform &IGCPlatform,
float profilingTimerResolution, const ShaderHash &inputShHash,
std::string &errorMessage);
// This function takes SPMD and ESIMD modules explicitly.
bool TranslateBuildSPMDAndESIMD(llvm::ArrayRef<SPVTranslationPair> inputModules,
TC::STB_TranslateOutputArgs *pOutputArgs, TC::TB_DATA_FORMAT inputDataFormatTemp,
const IGC::CPlatform &IGCPlatform, float profilingTimerResolution,
const ShaderHash &inputShHash, std::string &errorMessage);
} // namespace VLD
} // namespace IGC
|
