src/compiler/spirv/spirv2nir.c

   1 /*
   2  * Copyright © 2015 Intel Corporation
   3  *
   4  * Permission is hereby granted, free of charge, to any person obtaining a
   5  * copy of this software and associated documentation files (the "Software"),
   6  * to deal in the Software without restriction, including without limitation
   7  * the rights to use, copy, modify, merge, publish, distribute, sublicense,
   8  * and/or sell copies of the Software, and to permit persons to whom the
   9  * Software is furnished to do so, subject to the following conditions:
  10  *
  11  * The above copyright notice and this permission notice (including the next
  12  * paragraph) shall be included in all copies or substantial portions of the
  13  * Software.
  14  *
  15  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
  16  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
  17  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL
  18  * THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
  19  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING
  20  * FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS
  21  * IN THE SOFTWARE.
  22  *
  23  * Authors:
  24  *    Jason Ekstrand (jason@jlekstrand.net)
  25  *
  26  */
  27
  28 /*
  29  * A simple executable that opens a SPIR-V shader, converts it to NIR, and
  30  * dumps out the result.  This should be useful for testing the
  31  * spirv_to_nir code.
  32  */
  33
  34 #include "spirv/nir_spirv.h"
  35
  36 #include <sys/mman.h>
  37 #include <sys/types.h>
  38 #include <fcntl.h>
  39 #include <unistd.h>
  40 #include <stdio.h>
  41 #include <errno.h>
  42 #include <string.h>
  43 #include <getopt.h>
  44
  45 #define WORD_SIZE 4
  46
  47 static gl_shader_stage
  48 stage_to_enum(char *stage)
  49 {
  50    if (!strcmp(stage, "vertex"))
  51       return MESA_SHADER_VERTEX;
  52    else if (!strcmp(stage, "tess-ctrl"))
  53       return MESA_SHADER_TESS_CTRL;
  54    else if (!strcmp(stage, "tess-eval"))
  55       return MESA_SHADER_TESS_EVAL;
  56    else if (!strcmp(stage, "geometry"))
  57       return MESA_SHADER_GEOMETRY;
  58    else if (!strcmp(stage, "fragment"))
  59       return MESA_SHADER_FRAGMENT;
  60    else if (!strcmp(stage, "compute"))
  61       return MESA_SHADER_COMPUTE;
  62    else if (!strcmp(stage, "kernel"))
  63       return MESA_SHADER_KERNEL;
  64    else
  65       return MESA_SHADER_NONE;
  66 }
  67
  68 int main(int argc, char **argv)
  69 {
  70    gl_shader_stage shader_stage = MESA_SHADER_FRAGMENT;
  71    char *entry_point = "main";
  72    int ch;
  73
  74    static struct option long_options[] =
  75      {
  76        {"stage",  required_argument, 0, 's'},
  77        {"entry",  required_argument, 0, 'e'},
  78        {0, 0, 0, 0}
  79      };
  80
  81    while ((ch = getopt_long(argc - 1, argv + 1, "s:e:", long_options, NULL)) != -1)
  82    {
  83       switch (ch)
  84       {
  85          case 's':
  86             shader_stage = stage_to_enum(optarg);
  87             if (shader_stage == MESA_SHADER_NONE)
  88             {
  89                fprintf(stderr, "Unknown stage %s\n", optarg);
  90                return 1;
  91             }
  92             break;
  93          case 'e':
  94             entry_point = optarg;
  95             break;
  96          default:
  97             fprintf(stderr, "Unrecognized option.\n");
  98             return 1;
  99       }
 100    }
 101
 102    int fd = open(argv[1], O_RDONLY);
 103    if (fd < 0)
 104    {
 105       fprintf(stderr, "Failed to open %s\n", argv[1]);
 106       return 1;
 107    }
 108
 109    off_t len = lseek(fd, 0, SEEK_END);
 110    if (len % WORD_SIZE != 0)
 111    {
 112       fprintf(stderr, "File length isn't a multiple of the word size\n");
 113       fprintf(stderr, "Are you sure this is a valid SPIR-V shader?\n");
 114       close(fd);
 115       return 1;
 116    }
 117
 118    size_t word_count = len / WORD_SIZE;
 119
 120    const void *map = mmap(NULL, len, PROT_READ, MAP_PRIVATE, fd, 0);
 121    if (map == MAP_FAILED)
 122    {
 123       fprintf(stderr, "Failed to mmap the file: errno=%d, %s\n",
 124               errno, strerror(errno));
 125       close(fd);
 126       return 1;
 127    }
 128
 129    glsl_type_singleton_init_or_ref();
 130
 131    struct spirv_to_nir_options spirv_opts = {0};
 132    if (shader_stage == MESA_SHADER_KERNEL) {
 133       spirv_opts.environment = NIR_SPIRV_OPENCL;
 134       spirv_opts.caps.address = true;
 135       spirv_opts.caps.float64 = true;
 136       spirv_opts.caps.int8 = true;
 137       spirv_opts.caps.int16 = true;
 138       spirv_opts.caps.int64 = true;
 139       spirv_opts.caps.kernel = true;
 140    }
 141
 142    nir_shader *nir = spirv_to_nir(map, word_count, NULL, 0,
 143                                   shader_stage, entry_point,
 144                                   &spirv_opts, NULL);
 145
 146    if (nir)
 147       nir_print_shader(nir, stderr);
 148    else
 149       fprintf(stderr, "SPIRV to NIR compilation failed\n");
 150
 151    glsl_type_singleton_decref();
 152
 153    return 0;
 154 }