AlexIndustrial/mesa
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

nir/precompiled: add helper to emit an enum map for multiple targets

Browse Source 
Signed-off-by: Simon Perretta <simon.perretta@imgtec.com>
Reviewed-by: Alyssa Rosenzweig <alyssa@rosenzweig.io>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/36001>
This commit is contained in:
Simon Perretta
2024-12-24 12:37:07 +00:00
parent f118c6062a
commit 5b29daf7bc
1 changed files with 60 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/compiler/nir/nir_precompiled.h
+60
View File
@@ -6,6 +6,7 @@
#pragma once
#include <ctype.h>
#include <inttypes.h>
#include "nir.h"
#include "nir_builder.h"
#include "nir_serialize.h"
@@ -566,6 +567,65 @@ nir_precomp_print_binary_map(FILE *fp, const nir_shader *nir,
fprintf(fp, "};\n\n");
}
static inline void
nir_precomp_print_target_enum_map(FILE *fp_c, FILE *fp_h, const char *prefix, unsigned num_targets, const char **targets, uint64_t *target_ids)
{
/* Generate an enum indexing all devices */
fprintf(fp_h, "enum %s_target {\n", prefix);
for (unsigned t = 0; t < num_targets; ++t) {
fprintf(fp_h, " ");
nir_print_uppercase(fp_h, prefix);
fprintf(fp_h, "_TARGET_");
nir_print_uppercase(fp_h, targets[t]);
fprintf(fp_h, " = %u,\n", t);
}
fprintf(fp_h, " ");
nir_print_uppercase(fp_h, prefix);
fprintf(fp_h, "_NUM_TARGETS,\n");
fprintf(fp_h, "};\n");
if (!target_ids)
return;
fprintf(fp_h, "extern const uint64_t %s_target_id_map[", prefix);
nir_print_uppercase(fp_h, prefix);
fprintf(fp_h, "_NUM_TARGETS");
fprintf(fp_h, "];\n");
fprintf(fp_c, "const uint64_t %s_target_id_map[", prefix);
nir_print_uppercase(fp_c, prefix);
fprintf(fp_c, "_NUM_TARGETS");
fprintf(fp_c, "] = {\n");
for (unsigned t = 0; t < num_targets; ++t) {
fprintf(fp_c, " [");
nir_print_uppercase(fp_c, prefix);
fprintf(fp_c, "_TARGET_");
nir_print_uppercase(fp_c, targets[t]);
fprintf(fp_c, "] = 0x%" PRIx64 ",\n", target_ids[t]);
}
fprintf(fp_c, "};\n\n");
}
static inline void
nir_precomp_print_target_binary_map(FILE *fp_c, FILE *fp_h, const char *prefix, unsigned num_targets, const char **targets)
{
fprintf(fp_h, "extern const uint32_t **%s_targets[", prefix);
nir_print_uppercase(fp_h, prefix);
fprintf(fp_h, "_NUM_TARGETS];\n");
fprintf(fp_c, "const uint32_t **%s_targets[", prefix);
nir_print_uppercase(fp_c, prefix);
fprintf(fp_c, "_NUM_TARGETS] = {\n");
for (unsigned t = 0; t < num_targets; ++t) {
fprintf(fp_c, " [");
nir_print_uppercase(fp_c, prefix);
fprintf(fp_c, "_TARGET_");
nir_print_uppercase(fp_c, targets[t]);
fprintf(fp_c, "] = %s_%s,\n", prefix, targets[t]);
}
fprintf(fp_c, "};\n\n");
}
static inline nir_shader *
nir_precompiled_build_variant(const nir_function *libfunc, unsigned variant,
const nir_shader_compiler_options *opts,