frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

[vulkan] Add image format qualifier to glsl files (#69330)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/69330

 ---

## Context

Previously, our shader files did not declare any [image format qualifiers](https://www.khronos.org/opengl/wiki/Layout_Qualifier_(GLSL)#Image_formats) for image layouts. This causes the SPIR-V modules produced to declare the [StorageImageWriteWithoutFormat](https://www.khronos.org/registry/SPIR-V/specs/unified1/SPIRV.html#_a_id_capability_a_capability) capability, which requires `shaderStorageImageWriteWithoutFormat` to be enabled in [VkPhysicalDeviceFeatures](https://www.khronos.org/registry/vulkan/specs/1.2-extensions/man/html/VkPhysicalDeviceFeatures.html). `shaderStorageImageWriteWithoutFormat` is not available on some devices, causing errors to be reported by the Vulkan validation layer.

## Changes

Vulkan shaders now declare the image format explicitly so that the SPIR-V modules produced are compatible with devices that do not have `shaderStorageImageWriteWithoutFormat` enabled.

Test Plan: Imported from OSS

Reviewed By: beback4u

Differential Revision: D32840909

Pulled By: SS-JIA

fbshipit-source-id: 76e0a0da68b423ebc74ae7e839b9cfaf57d2cd39
This commit is contained in:
Sicheng Stephen Jia
2021-12-07 16:21:18 -08:00
committed by Facebook GitHub Bot
parent e5a1ee0e5a
commit bede33e3f5
53 changed files with 218 additions and 164 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
aten/src/ATen/gen_vulkan_glsl.py
View File

@ -7,7 +7,7 @@ from tools.codegen.code_template import CodeTemplate
H_NAME = "glsl.h" H_NAME = "glsl.h"
CPP_NAME = "glsl.cpp" CPP_NAME = "glsl.cpp"
DEFAULT_ENV = {"precision": "highp"} DEFAULT_ENV = {"precision": "highp", "format": "rgba32f"}
def findAllGlsls(path): def findAllGlsls(path):
cmd = "find " + path + " -name \"*.glsl\"" cmd = "find " + path + " -name \"*.glsl\""

2
aten/src/ATen/gen_vulkan_spv.py
View File

@ -9,7 +9,7 @@ from tools.codegen.code_template import CodeTemplate
H_NAME = "spv.h" H_NAME = "spv.h"
CPP_NAME = "spv.cpp" CPP_NAME = "spv.cpp"
DEFAULT_ENV = {"precision": "highp"} DEFAULT_ENV = {"precision": "highp", "format": "rgba32f"}
def getName(filePath): def getName(filePath):
return os.path.basename(filePath).replace("/", "_").replace(".", "_") return os.path.basename(filePath).replace("/", "_").replace(".", "_")

7
aten/src/ATen/native/vulkan/glsl/adaptive_avg_pool2d.glsl
View File

@ -1,13 +1,14 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0) uniform PRECISION restrict writeonly image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict writeonly image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput;
layout(set = 0, binding = 2) uniform PRECISION restrict Block { layout(set = 0, binding = 2) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
vec2 kernel; vec2 kernel;
vec2 stride; vec2 stride;

9
aten/src/ATen/native/vulkan/glsl/add.glsl
View File

@ -1,14 +1,15 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0) uniform PRECISION restrict writeonly image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict writeonly image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput0; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput0;
layout(set = 0, binding = 2) uniform PRECISION sampler3D uInput1; layout(set = 0, binding = 2) uniform PRECISION sampler3D uInput1;
layout(set = 0, binding = 3) uniform PRECISION restrict Block { layout(set = 0, binding = 3) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
ivec4 isize0; ivec4 isize0;
ivec3 isize1; ivec3 isize1;

7
aten/src/ATen/native/vulkan/glsl/add_.glsl
View File

@ -1,13 +1,14 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0, rgba16f) uniform PRECISION restrict image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput0; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput0;
layout(set = 0, binding = 2) uniform PRECISION restrict Block { layout(set = 0, binding = 2) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
ivec3 isize; ivec3 isize;
float alpha; float alpha;

7
aten/src/ATen/native/vulkan/glsl/add_scalar.glsl
View File

@ -1,13 +1,14 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0) uniform PRECISION restrict writeonly image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict writeonly image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput;
layout(set = 0, binding = 2) uniform PRECISION restrict Block { layout(set = 0, binding = 2) uniform PRECISION restrict Block {
ivec3 size; ivec3 size;
float other; float other;
} uBlock; } uBlock;

5
aten/src/ATen/native/vulkan/glsl/add_scalar_.glsl
View File

@ -1,12 +1,13 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0, rgba16f) uniform PRECISION restrict image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION restrict Block { layout(set = 0, binding = 1) uniform PRECISION restrict Block {
ivec3 size; ivec3 size;
float other; float other;
} uBlock; } uBlock;

11
aten/src/ATen/native/vulkan/glsl/addmm.glsl
View File

@ -1,15 +1,16 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0) uniform PRECISION restrict writeonly image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict writeonly image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uM1; layout(set = 0, binding = 1) uniform PRECISION sampler3D uM1;
layout(set = 0, binding = 2) uniform PRECISION sampler3D uM2; layout(set = 0, binding = 2) uniform PRECISION sampler3D uM2;
layout(set = 0, binding = 3) uniform PRECISION sampler3D uT; layout(set = 0, binding = 3) uniform PRECISION sampler3D uT;
layout(set = 0, binding = 4) uniform PRECISION restrict Block { layout(set = 0, binding = 4) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
vec2 multiplier; vec2 multiplier;
} uBlock; } uBlock;

7
aten/src/ATen/native/vulkan/glsl/avg_pool2d.glsl
View File

@ -1,13 +1,14 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0) uniform PRECISION restrict writeonly image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict writeonly image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput;
layout(set = 0, binding = 2) uniform PRECISION restrict Block { layout(set = 0, binding = 2) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
ivec4 kernel; ivec4 kernel;
ivec2 stride; ivec2 stride;

7
aten/src/ATen/native/vulkan/glsl/cat_feature.glsl
View File

@ -1,13 +1,14 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0, rgba16f) uniform PRECISION image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput;
layout(set = 0, binding = 2) uniform PRECISION restrict Block { layout(set = 0, binding = 2) uniform PRECISION restrict Block {
ivec4 size; // output texture size (x=width,y=height,z=depth,w=unused) ivec4 size; // output texture size (x=width,y=height,z=depth,w=unused)
ivec4 isize; // input texture size (x=width,y=height,z=depth,w=unused) ivec4 isize; // input texture size (x=width,y=height,z=depth,w=unused)
uint batch_size; // input tensor's batch size uint batch_size; // input tensor's batch size

7
aten/src/ATen/native/vulkan/glsl/clamp.glsl
View File

@ -1,13 +1,14 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0) uniform PRECISION restrict writeonly image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict writeonly image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput;
layout(set = 0, binding = 2) uniform PRECISION restrict Block { layout(set = 0, binding = 2) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
vec2 clamp; vec2 clamp;
} uBlock; } uBlock;

5
aten/src/ATen/native/vulkan/glsl/clamp_.glsl
View File

@ -1,12 +1,13 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0, rgba16f) uniform PRECISION restrict image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION restrict Block { layout(set = 0, binding = 1) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
vec2 clamp; vec2 clamp;
} uBlock; } uBlock;

11
aten/src/ATen/native/vulkan/glsl/conv2d.glsl
View File

@ -1,15 +1,16 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0) uniform PRECISION restrict writeonly image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict writeonly image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput;
layout(set = 0, binding = 2) uniform PRECISION sampler3D uKernel; layout(set = 0, binding = 2) uniform PRECISION sampler3D uKernel;
layout(set = 0, binding = 3) uniform PRECISION sampler3D uBias; layout(set = 0, binding = 3) uniform PRECISION sampler3D uBias;
layout(set = 0, binding = 4) uniform PRECISION restrict Block { layout(set = 0, binding = 4) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
ivec4 kernel; ivec4 kernel;
ivec2 ikernel; ivec2 ikernel;

11
aten/src/ATen/native/vulkan/glsl/conv2d_dw.glsl
View File

@ -1,15 +1,16 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0) uniform PRECISION restrict writeonly image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict writeonly image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput;
layout(set = 0, binding = 2) uniform PRECISION sampler3D uKernel; layout(set = 0, binding = 2) uniform PRECISION sampler3D uKernel;
layout(set = 0, binding = 3) uniform PRECISION sampler3D uBias; layout(set = 0, binding = 3) uniform PRECISION sampler3D uBias;
layout(set = 0, binding = 4) uniform PRECISION restrict Block { layout(set = 0, binding = 4) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
ivec4 kernel; ivec4 kernel;
ivec2 ikernel; ivec2 ikernel;

10
aten/src/ATen/native/vulkan/glsl/conv2d_dw_clamp.glsl
View File

@ -1,10 +1,12 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
layout(set = 0, rgba16f, binding = 0) writeonly PRECISION uniform image3D uOutput; layout(set = 0, binding = 0, FORMAT) writeonly PRECISION uniform image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput;
layout(set = 0, binding = 2) uniform PRECISION sampler3D uKernel; layout(set = 0, binding = 2) uniform PRECISION sampler3D uKernel;
layout(set = 0, binding = 3) readonly buffer bias { layout(set = 0, binding = 3) readonly buffer bias {
vec4 data[]; vec4 data[];
} }
uBias; uBias;

11
aten/src/ATen/native/vulkan/glsl/conv2d_pw.glsl
View File

@ -1,15 +1,16 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0) uniform PRECISION restrict writeonly image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict writeonly image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput;
layout(set = 0, binding = 2) uniform PRECISION sampler3D uKernel; layout(set = 0, binding = 2) uniform PRECISION sampler3D uKernel;
layout(set = 0, binding = 3) uniform PRECISION sampler3D uBias; layout(set = 0, binding = 3) uniform PRECISION sampler3D uBias;
layout(set = 0, binding = 4) uniform PRECISION restrict Block { layout(set = 0, binding = 4) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
ivec4 kernel; ivec4 kernel;
ivec2 ikernel; ivec2 ikernel;

11
aten/src/ATen/native/vulkan/glsl/conv2d_pw_2x2.glsl
View File

@ -1,15 +1,16 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0) uniform PRECISION restrict writeonly image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict writeonly image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput;
layout(set = 0, binding = 2) uniform PRECISION sampler3D uKernel; layout(set = 0, binding = 2) uniform PRECISION sampler3D uKernel;
layout(set = 0, binding = 3) uniform PRECISION sampler3D uBias; layout(set = 0, binding = 3) uniform PRECISION sampler3D uBias;
layout(set = 0, binding = 4) uniform PRECISION restrict Block { layout(set = 0, binding = 4) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
ivec4 kernel; ivec4 kernel;
ivec2 ikernel; ivec2 ikernel;

11
aten/src/ATen/native/vulkan/glsl/conv2d_pw_2x2_buffered.glsl
View File

@ -1,15 +1,16 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0) uniform PRECISION restrict writeonly image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict writeonly image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput;
layout(set = 0, binding = 2) uniform PRECISION sampler3D uKernel; layout(set = 0, binding = 2) uniform PRECISION sampler3D uKernel;
layout(set = 0, binding = 3) uniform PRECISION sampler3D uBias; layout(set = 0, binding = 3) uniform PRECISION sampler3D uBias;
layout(set = 0, binding = 4) uniform PRECISION restrict Block { layout(set = 0, binding = 4) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
ivec4 kernel; ivec4 kernel;
ivec2 ikernel; ivec2 ikernel;

9
aten/src/ATen/native/vulkan/glsl/conv2d_winograd_2_3.glsl
View File

@ -1,14 +1,15 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0) uniform PRECISION restrict writeonly image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict writeonly image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput;
layout(set = 0, binding = 2) uniform PRECISION sampler3D uKernel; layout(set = 0, binding = 2) uniform PRECISION sampler3D uKernel;
layout(set = 0, binding = 3) buffer PRECISION restrict readonly Bias { layout(set = 0, binding = 3) buffer PRECISION restrict readonly Bias {
vec4 data[]; vec4 data[];
} uBias; } uBias;
layout(set = 0, binding = 4) uniform PRECISION restrict Block { layout(set = 0, binding = 4) uniform PRECISION restrict Block {

11
aten/src/ATen/native/vulkan/glsl/conv_transpose2d.glsl
View File

@ -1,15 +1,16 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0) uniform PRECISION restrict writeonly image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict writeonly image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput;
layout(set = 0, binding = 2) uniform PRECISION sampler3D uKernel; layout(set = 0, binding = 2) uniform PRECISION sampler3D uKernel;
layout(set = 0, binding = 3) uniform PRECISION sampler3D uBias; layout(set = 0, binding = 3) uniform PRECISION sampler3D uBias;
layout(set = 0, binding = 4) uniform PRECISION restrict Block { layout(set = 0, binding = 4) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
ivec4 kernel; ivec4 kernel;
ivec2 ikernel; ivec2 ikernel;

9
aten/src/ATen/native/vulkan/glsl/div.glsl
View File

@ -1,14 +1,15 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0) uniform PRECISION restrict writeonly image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict writeonly image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput0; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput0;
layout(set = 0, binding = 2) uniform PRECISION sampler3D uInput1; layout(set = 0, binding = 2) uniform PRECISION sampler3D uInput1;
layout(set = 0, binding = 3) uniform PRECISION restrict Block { layout(set = 0, binding = 3) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
ivec4 isize0; ivec4 isize0;
ivec3 isize1; ivec3 isize1;

7
aten/src/ATen/native/vulkan/glsl/div_.glsl
View File

@ -1,13 +1,14 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0, rgba16f) uniform PRECISION restrict image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput0; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput0;
layout(set = 0, binding = 2) uniform PRECISION restrict Block { layout(set = 0, binding = 2) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
ivec3 isize; ivec3 isize;
float alpha; float alpha;

7
aten/src/ATen/native/vulkan/glsl/hardshrink.glsl
View File

@ -1,13 +1,14 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0) uniform PRECISION restrict writeonly image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict writeonly image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput;
layout(set = 0, binding = 2) uniform PRECISION restrict Block { layout(set = 0, binding = 2) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
float lambd; float lambd;
} uBlock; } uBlock;

5
aten/src/ATen/native/vulkan/glsl/hardshrink_.glsl
View File

@ -1,12 +1,13 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0, rgba16f) uniform PRECISION restrict image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION restrict Block { layout(set = 0, binding = 1) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
float lambd; float lambd;
} uBlock; } uBlock;

7
aten/src/ATen/native/vulkan/glsl/hardsigmoid.glsl
View File

@ -1,13 +1,14 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0) uniform PRECISION restrict writeonly image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict writeonly image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput;
layout(set = 0, binding = 2) uniform PRECISION restrict Block { layout(set = 0, binding = 2) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
} uBlock; } uBlock;

5
aten/src/ATen/native/vulkan/glsl/hardsigmoid_.glsl
View File

@ -1,12 +1,13 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0, rgba16f) uniform PRECISION restrict image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION restrict Block { layout(set = 0, binding = 1) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
} uBlock; } uBlock;

7
aten/src/ATen/native/vulkan/glsl/hardswish.glsl
View File

@ -1,13 +1,14 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0) uniform PRECISION restrict writeonly image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict writeonly image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput;
layout(set = 0, binding = 2) uniform PRECISION restrict Block { layout(set = 0, binding = 2) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
} uBlock; } uBlock;

5
aten/src/ATen/native/vulkan/glsl/hardswish_.glsl
View File

@ -1,12 +1,13 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0, rgba16f) uniform PRECISION restrict image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION restrict Block { layout(set = 0, binding = 1) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
} uBlock; } uBlock;

7
aten/src/ATen/native/vulkan/glsl/leaky_relu.glsl
View File

@ -1,13 +1,14 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0) uniform PRECISION restrict writeonly image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict writeonly image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput;
layout(set = 0, binding = 2) uniform PRECISION restrict Block { layout(set = 0, binding = 2) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
float negative_slope; float negative_slope;
} uBlock; } uBlock;

5
aten/src/ATen/native/vulkan/glsl/leaky_relu_.glsl
View File

@ -1,12 +1,13 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0, rgba16f) uniform PRECISION restrict image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION restrict Block { layout(set = 0, binding = 1) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
float negative_slope; float negative_slope;
} uBlock; } uBlock;

7
aten/src/ATen/native/vulkan/glsl/log_softmax.glsl
View File

@ -1,13 +1,14 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0) uniform PRECISION restrict writeonly image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict writeonly image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput;
layout(set = 0, binding = 2) uniform PRECISION restrict Block { layout(set = 0, binding = 2) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
} uBlock; } uBlock;

7
aten/src/ATen/native/vulkan/glsl/max_pool2d.glsl
View File

@ -1,13 +1,14 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0) uniform PRECISION restrict writeonly image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict writeonly image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput;
layout(set = 0, binding = 2) uniform PRECISION restrict Block { layout(set = 0, binding = 2) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
ivec4 kernel; ivec4 kernel;
ivec2 stride; ivec2 stride;

7
aten/src/ATen/native/vulkan/glsl/mean.glsl
View File

@ -1,13 +1,14 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0) uniform PRECISION restrict writeonly image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict writeonly image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput;
layout(set = 0, binding = 2) uniform PRECISION restrict Block { layout(set = 0, binding = 2) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
ivec3 isize; ivec3 isize;
} uBlock; } uBlock;

7
aten/src/ATen/native/vulkan/glsl/mean2d.glsl
View File

@ -1,13 +1,14 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0) uniform PRECISION restrict writeonly image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict writeonly image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput;
layout(set = 0, binding = 2) uniform PRECISION restrict Block { layout(set = 0, binding = 2) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
ivec3 isize; ivec3 isize;
} uBlock; } uBlock;

9
aten/src/ATen/native/vulkan/glsl/mm.glsl
View File

@ -1,14 +1,15 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0) uniform PRECISION restrict writeonly image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict writeonly image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uM1; layout(set = 0, binding = 1) uniform PRECISION sampler3D uM1;
layout(set = 0, binding = 2) uniform PRECISION sampler3D uM2; layout(set = 0, binding = 2) uniform PRECISION sampler3D uM2;
layout(set = 0, binding = 3) uniform PRECISION restrict Block { layout(set = 0, binding = 3) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
} uBlock; } uBlock;

9
aten/src/ATen/native/vulkan/glsl/mul.glsl
View File

@ -1,14 +1,15 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0) uniform PRECISION restrict writeonly image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict writeonly image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput0; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput0;
layout(set = 0, binding = 2) uniform PRECISION sampler3D uInput1; layout(set = 0, binding = 2) uniform PRECISION sampler3D uInput1;
layout(set = 0, binding = 3) uniform PRECISION restrict Block { layout(set = 0, binding = 3) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
ivec4 isize0; ivec4 isize0;
ivec3 isize1; ivec3 isize1;

7
aten/src/ATen/native/vulkan/glsl/mul_.glsl
View File

@ -1,13 +1,14 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0, rgba16f) uniform PRECISION restrict image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput0; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput0;
layout(set = 0, binding = 2) uniform PRECISION restrict Block { layout(set = 0, binding = 2) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
ivec3 isize; ivec3 isize;
float alpha; float alpha;

7
aten/src/ATen/native/vulkan/glsl/mul_scalar.glsl
View File

@ -1,13 +1,14 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0) uniform PRECISION restrict writeonly image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict writeonly image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput;
layout(set = 0, binding = 2) uniform PRECISION restrict Block { layout(set = 0, binding = 2) uniform PRECISION restrict Block {
ivec3 size; ivec3 size;
float other; float other;
} uBlock; } uBlock;

5
aten/src/ATen/native/vulkan/glsl/mul_scalar_.glsl
View File

@ -1,12 +1,13 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0, rgba16f) uniform PRECISION restrict image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION restrict Block { layout(set = 0, binding = 1) uniform PRECISION restrict Block {
ivec3 size; ivec3 size;
float other; float other;
} uBlock; } uBlock;

7
aten/src/ATen/native/vulkan/glsl/nchw_to_image.glsl
View File

@ -1,15 +1,16 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0) uniform PRECISION restrict writeonly image3D uImage; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict writeonly image3D uImage;
layout(set = 0, binding = 1) buffer PRECISION restrict readonly Buffer { layout(set = 0, binding = 1) buffer PRECISION restrict readonly Buffer {
float data[]; float data[];
} uBuffer; } uBuffer;
layout(set = 0, binding = 2) uniform PRECISION restrict Block { layout(set = 0, binding = 2) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
ivec4 offset; ivec4 offset;
} uBlock; } uBlock;

7
aten/src/ATen/native/vulkan/glsl/permute_4d.glsl
View File

@ -1,13 +1,14 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0, rgba16f) uniform PRECISION image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput;
layout(set = 0, binding = 2) uniform PRECISION restrict Block { layout(set = 0, binding = 2) uniform PRECISION restrict Block {
ivec4 size; // output texture size (x=width,y=height,z=depth,w=unused) ivec4 size; // output texture size (x=width,y=height,z=depth,w=unused)
ivec4 isize; // input texture size (x=width,y=height,z=depth,w=unused) ivec4 isize; // input texture size (x=width,y=height,z=depth,w=unused)
uvec4 tensor_size; // output tensor size uvec4 tensor_size; // output tensor size

7
aten/src/ATen/native/vulkan/glsl/reflection_pad2d.glsl
View File

@ -1,13 +1,14 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0) uniform PRECISION restrict writeonly image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict writeonly image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput;
layout(set = 0, binding = 2) uniform PRECISION restrict Block { layout(set = 0, binding = 2) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
ivec4 pad; ivec4 pad;
} uBlock; } uBlock;

7
aten/src/ATen/native/vulkan/glsl/sigmoid.glsl
View File

@ -1,13 +1,14 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0) uniform PRECISION restrict writeonly image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict writeonly image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput;
layout(set = 0, binding = 2) uniform PRECISION restrict Block { layout(set = 0, binding = 2) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
} uBlock; } uBlock;

5
aten/src/ATen/native/vulkan/glsl/sigmoid_.glsl
View File

@ -1,12 +1,13 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0, rgba16f) uniform PRECISION restrict image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION restrict Block { layout(set = 0, binding = 1) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
} uBlock; } uBlock;

7
aten/src/ATen/native/vulkan/glsl/slice_4d.glsl
View File

@ -1,13 +1,14 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0, rgba16f) uniform PRECISION image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput;
layout(set = 0, binding = 2) uniform PRECISION restrict Block { layout(set = 0, binding = 2) uniform PRECISION restrict Block {
ivec4 size; // output texture size (x=width,y=height,z=depth,w=unused) ivec4 size; // output texture size (x=width,y=height,z=depth,w=unused)
ivec4 isize; // input texture size (x=width,y=height,z=depth,w=unused) ivec4 isize; // input texture size (x=width,y=height,z=depth,w=unused)
uvec4 tensor_size; // output tensor size uvec4 tensor_size; // output tensor size

7
aten/src/ATen/native/vulkan/glsl/softmax.glsl
View File

@ -1,13 +1,14 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0) uniform PRECISION restrict writeonly image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict writeonly image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput;
layout(set = 0, binding = 2) uniform PRECISION restrict Block { layout(set = 0, binding = 2) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
} uBlock; } uBlock;

9
aten/src/ATen/native/vulkan/glsl/sub.glsl
View File

@ -1,14 +1,15 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0) uniform PRECISION restrict writeonly image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict writeonly image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput0; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput0;
layout(set = 0, binding = 2) uniform PRECISION sampler3D uInput1; layout(set = 0, binding = 2) uniform PRECISION sampler3D uInput1;
layout(set = 0, binding = 3) uniform PRECISION restrict Block { layout(set = 0, binding = 3) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
ivec4 isize0; ivec4 isize0;
ivec3 isize1; ivec3 isize1;

7
aten/src/ATen/native/vulkan/glsl/sub_.glsl
View File

@ -1,13 +1,14 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0, rgba16f) uniform PRECISION restrict image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput0; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput0;
layout(set = 0, binding = 2) uniform PRECISION restrict Block { layout(set = 0, binding = 2) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
ivec3 isize; ivec3 isize;
float alpha; float alpha;

7
aten/src/ATen/native/vulkan/glsl/tanh.glsl
View File

@ -1,13 +1,14 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0) uniform PRECISION restrict writeonly image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict writeonly image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput;
layout(set = 0, binding = 2) uniform PRECISION restrict Block { layout(set = 0, binding = 2) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
} uBlock; } uBlock;

5
aten/src/ATen/native/vulkan/glsl/tanh_.glsl
View File

@ -1,12 +1,13 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0, rgba16f) uniform PRECISION restrict image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION restrict Block { layout(set = 0, binding = 1) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
} uBlock; } uBlock;

7
aten/src/ATen/native/vulkan/glsl/transform_winograd_2_3_sh.glsl
View File

@ -1,13 +1,14 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0) uniform PRECISION restrict writeonly image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict writeonly image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput;
layout(set = 0, binding = 2) uniform PRECISION restrict Block { layout(set = 0, binding = 2) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
ivec2 limits; ivec2 limits;
ivec2 padding; ivec2 padding;

7
aten/src/ATen/native/vulkan/glsl/upsample_nearest2d.glsl
View File

@ -1,13 +1,14 @@
#version 450 core #version 450 core
#define PRECISION $precision #define PRECISION $precision
#define FORMAT $format
layout(std430) buffer; layout(std430) buffer;
/* Qualifiers: layout - storage - precision - memory */ /* Qualifiers: layout - storage - precision - memory */
layout(set = 0, binding = 0) uniform PRECISION restrict writeonly image3D uOutput; layout(set = 0, binding = 0, FORMAT) uniform PRECISION restrict writeonly image3D uOutput;
layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput; layout(set = 0, binding = 1) uniform PRECISION sampler3D uInput;
layout(set = 0, binding = 2) uniform PRECISION restrict Block { layout(set = 0, binding = 2) uniform PRECISION restrict Block {
ivec4 size; ivec4 size;
ivec2 isize; ivec2 isize;
vec2 scale; vec2 scale;

3
cmake/VulkanCodegen.cmake
View File

@ -9,6 +9,9 @@ set(VULKAN_GEN_ARG_ENV "")
if(USE_VULKAN_RELAXED_PRECISION) if(USE_VULKAN_RELAXED_PRECISION)
string(APPEND VULKAN_GEN_ARG_ENV "precision=mediump") string(APPEND VULKAN_GEN_ARG_ENV "precision=mediump")
endif() endif()
if(USE_VULKAN_FP16_INFERENCE)
string(APPEND VULKAN_GEN_ARG_ENV "format=rgba16f")
endif()
if(USE_VULKAN_SHADERC_RUNTIME) if(USE_VULKAN_SHADERC_RUNTIME)
set(PYTHONPATH "$ENV{PYTHONPATH}") set(PYTHONPATH "$ENV{PYTHONPATH}")