-
Notifications
You must be signed in to change notification settings - Fork 562
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Merge pull request #2326 from KhronosGroup/pr-2325
Land PR 2325.
- Loading branch information
Showing
10 changed files
with
489 additions
and
9 deletions.
There are no files selected for viewing
90 changes: 90 additions & 0 deletions
90
reference/opt/shaders-msl/frag/gather-compare-const-offsets.frag
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,90 @@ | ||
#pragma clang diagnostic ignored "-Wmissing-prototypes" | ||
#pragma clang diagnostic ignored "-Wmissing-braces" | ||
|
||
#include <metal_stdlib> | ||
#include <simd/simd.h> | ||
|
||
using namespace metal; | ||
|
||
template<typename T, size_t Num> | ||
struct spvUnsafeArray | ||
{ | ||
T elements[Num ? Num : 1]; | ||
|
||
thread T& operator [] (size_t pos) thread | ||
{ | ||
return elements[pos]; | ||
} | ||
constexpr const thread T& operator [] (size_t pos) const thread | ||
{ | ||
return elements[pos]; | ||
} | ||
|
||
device T& operator [] (size_t pos) device | ||
{ | ||
return elements[pos]; | ||
} | ||
constexpr const device T& operator [] (size_t pos) const device | ||
{ | ||
return elements[pos]; | ||
} | ||
|
||
constexpr const constant T& operator [] (size_t pos) const constant | ||
{ | ||
return elements[pos]; | ||
} | ||
|
||
threadgroup T& operator [] (size_t pos) threadgroup | ||
{ | ||
return elements[pos]; | ||
} | ||
constexpr const threadgroup T& operator [] (size_t pos) const threadgroup | ||
{ | ||
return elements[pos]; | ||
} | ||
}; | ||
|
||
template<typename T> struct spvRemoveReference { typedef T type; }; | ||
template<typename T> struct spvRemoveReference<thread T&> { typedef T type; }; | ||
template<typename T> struct spvRemoveReference<thread T&&> { typedef T type; }; | ||
template<typename T> inline constexpr thread T&& spvForward(thread typename spvRemoveReference<T>::type& x) | ||
{ | ||
return static_cast<thread T&&>(x); | ||
} | ||
template<typename T> inline constexpr thread T&& spvForward(thread typename spvRemoveReference<T>::type&& x) | ||
{ | ||
return static_cast<thread T&&>(x); | ||
} | ||
|
||
// Wrapper function that processes a texture gather with a constant offset array. | ||
template<typename T, template<typename, access = access::sample, typename = void> class Tex, typename Toff, typename... Tp> | ||
inline vec<T, 4> spvGatherCompareConstOffsets(const thread Tex<T>& t, sampler s, Toff coffsets, Tp... params) | ||
{ | ||
vec<T, 4> rslts[4]; | ||
for (uint i = 0; i < 4; i++) | ||
{ | ||
rslts[i] = t.gather_compare(s, spvForward<Tp>(params)..., coffsets[i]); | ||
} | ||
return vec<T, 4>(rslts[0].w, rslts[1].w, rslts[2].w, rslts[3].w); | ||
} | ||
|
||
constant spvUnsafeArray<int2, 4> _38 = spvUnsafeArray<int2, 4>({ int2(-8, 3), int2(-4, 7), int2(0, 3), int2(3, 0) }); | ||
|
||
struct main0_out | ||
{ | ||
float4 FragColor [[color(0)]]; | ||
}; | ||
|
||
struct main0_in | ||
{ | ||
float2 coord [[user(locn0)]]; | ||
float2 compare_value [[user(locn1)]]; | ||
}; | ||
|
||
fragment main0_out main0(main0_in in [[stage_in]], depth2d<float> tex [[texture(0)]], sampler texSmplr [[sampler(0)]]) | ||
{ | ||
main0_out out = {}; | ||
out.FragColor = spvGatherCompareConstOffsets(tex, texSmplr, _38, in.coord, in.compare_value.x); | ||
return out; | ||
} | ||
|
103 changes: 103 additions & 0 deletions
103
reference/opt/shaders-msl/frag/gather-const-offsets.frag
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,103 @@ | ||
#pragma clang diagnostic ignored "-Wmissing-prototypes" | ||
#pragma clang diagnostic ignored "-Wmissing-braces" | ||
|
||
#include <metal_stdlib> | ||
#include <simd/simd.h> | ||
|
||
using namespace metal; | ||
|
||
template<typename T, size_t Num> | ||
struct spvUnsafeArray | ||
{ | ||
T elements[Num ? Num : 1]; | ||
|
||
thread T& operator [] (size_t pos) thread | ||
{ | ||
return elements[pos]; | ||
} | ||
constexpr const thread T& operator [] (size_t pos) const thread | ||
{ | ||
return elements[pos]; | ||
} | ||
|
||
device T& operator [] (size_t pos) device | ||
{ | ||
return elements[pos]; | ||
} | ||
constexpr const device T& operator [] (size_t pos) const device | ||
{ | ||
return elements[pos]; | ||
} | ||
|
||
constexpr const constant T& operator [] (size_t pos) const constant | ||
{ | ||
return elements[pos]; | ||
} | ||
|
||
threadgroup T& operator [] (size_t pos) threadgroup | ||
{ | ||
return elements[pos]; | ||
} | ||
constexpr const threadgroup T& operator [] (size_t pos) const threadgroup | ||
{ | ||
return elements[pos]; | ||
} | ||
}; | ||
|
||
template<typename T> struct spvRemoveReference { typedef T type; }; | ||
template<typename T> struct spvRemoveReference<thread T&> { typedef T type; }; | ||
template<typename T> struct spvRemoveReference<thread T&&> { typedef T type; }; | ||
template<typename T> inline constexpr thread T&& spvForward(thread typename spvRemoveReference<T>::type& x) | ||
{ | ||
return static_cast<thread T&&>(x); | ||
} | ||
template<typename T> inline constexpr thread T&& spvForward(thread typename spvRemoveReference<T>::type&& x) | ||
{ | ||
return static_cast<thread T&&>(x); | ||
} | ||
|
||
// Wrapper function that processes a texture gather with a constant offset array. | ||
template<typename T, template<typename, access = access::sample, typename = void> class Tex, typename Toff, typename... Tp> | ||
inline vec<T, 4> spvGatherConstOffsets(const thread Tex<T>& t, sampler s, Toff coffsets, component c, Tp... params) METAL_CONST_ARG(c) | ||
{ | ||
vec<T, 4> rslts[4]; | ||
for (uint i = 0; i < 4; i++) | ||
{ | ||
switch (c) | ||
{ | ||
case component::x: | ||
rslts[i] = t.gather(s, spvForward<Tp>(params)..., coffsets[i], component::x); | ||
break; | ||
case component::y: | ||
rslts[i] = t.gather(s, spvForward<Tp>(params)..., coffsets[i], component::y); | ||
break; | ||
case component::z: | ||
rslts[i] = t.gather(s, spvForward<Tp>(params)..., coffsets[i], component::z); | ||
break; | ||
case component::w: | ||
rslts[i] = t.gather(s, spvForward<Tp>(params)..., coffsets[i], component::w); | ||
break; | ||
} | ||
} | ||
return vec<T, 4>(rslts[0].w, rslts[1].w, rslts[2].w, rslts[3].w); | ||
} | ||
|
||
constant spvUnsafeArray<int2, 4> _30 = spvUnsafeArray<int2, 4>({ int2(-8), int2(-8, 7), int2(7, -8), int2(7) }); | ||
|
||
struct main0_out | ||
{ | ||
float4 FragColor [[color(0)]]; | ||
}; | ||
|
||
struct main0_in | ||
{ | ||
float3 coord [[user(locn0)]]; | ||
}; | ||
|
||
fragment main0_out main0(main0_in in [[stage_in]], texture2d_array<float> tex [[texture(0)]], sampler texSmplr [[sampler(0)]]) | ||
{ | ||
main0_out out = {}; | ||
out.FragColor = spvGatherConstOffsets(tex, texSmplr, _30, component::y, in.coord.xy, uint(rint(in.coord.z))); | ||
return out; | ||
} | ||
|
90 changes: 90 additions & 0 deletions
90
reference/shaders-msl/frag/gather-compare-const-offsets.frag
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,90 @@ | ||
#pragma clang diagnostic ignored "-Wmissing-prototypes" | ||
#pragma clang diagnostic ignored "-Wmissing-braces" | ||
|
||
#include <metal_stdlib> | ||
#include <simd/simd.h> | ||
|
||
using namespace metal; | ||
|
||
template<typename T, size_t Num> | ||
struct spvUnsafeArray | ||
{ | ||
T elements[Num ? Num : 1]; | ||
|
||
thread T& operator [] (size_t pos) thread | ||
{ | ||
return elements[pos]; | ||
} | ||
constexpr const thread T& operator [] (size_t pos) const thread | ||
{ | ||
return elements[pos]; | ||
} | ||
|
||
device T& operator [] (size_t pos) device | ||
{ | ||
return elements[pos]; | ||
} | ||
constexpr const device T& operator [] (size_t pos) const device | ||
{ | ||
return elements[pos]; | ||
} | ||
|
||
constexpr const constant T& operator [] (size_t pos) const constant | ||
{ | ||
return elements[pos]; | ||
} | ||
|
||
threadgroup T& operator [] (size_t pos) threadgroup | ||
{ | ||
return elements[pos]; | ||
} | ||
constexpr const threadgroup T& operator [] (size_t pos) const threadgroup | ||
{ | ||
return elements[pos]; | ||
} | ||
}; | ||
|
||
template<typename T> struct spvRemoveReference { typedef T type; }; | ||
template<typename T> struct spvRemoveReference<thread T&> { typedef T type; }; | ||
template<typename T> struct spvRemoveReference<thread T&&> { typedef T type; }; | ||
template<typename T> inline constexpr thread T&& spvForward(thread typename spvRemoveReference<T>::type& x) | ||
{ | ||
return static_cast<thread T&&>(x); | ||
} | ||
template<typename T> inline constexpr thread T&& spvForward(thread typename spvRemoveReference<T>::type&& x) | ||
{ | ||
return static_cast<thread T&&>(x); | ||
} | ||
|
||
// Wrapper function that processes a texture gather with a constant offset array. | ||
template<typename T, template<typename, access = access::sample, typename = void> class Tex, typename Toff, typename... Tp> | ||
inline vec<T, 4> spvGatherCompareConstOffsets(const thread Tex<T>& t, sampler s, Toff coffsets, Tp... params) | ||
{ | ||
vec<T, 4> rslts[4]; | ||
for (uint i = 0; i < 4; i++) | ||
{ | ||
rslts[i] = t.gather_compare(s, spvForward<Tp>(params)..., coffsets[i]); | ||
} | ||
return vec<T, 4>(rslts[0].w, rslts[1].w, rslts[2].w, rslts[3].w); | ||
} | ||
|
||
constant spvUnsafeArray<int2, 4> _38 = spvUnsafeArray<int2, 4>({ int2(-8, 3), int2(-4, 7), int2(0, 3), int2(3, 0) }); | ||
|
||
struct main0_out | ||
{ | ||
float4 FragColor [[color(0)]]; | ||
}; | ||
|
||
struct main0_in | ||
{ | ||
float2 coord [[user(locn0)]]; | ||
float2 compare_value [[user(locn1)]]; | ||
}; | ||
|
||
fragment main0_out main0(main0_in in [[stage_in]], depth2d<float> tex [[texture(0)]], sampler texSmplr [[sampler(0)]]) | ||
{ | ||
main0_out out = {}; | ||
out.FragColor = spvGatherCompareConstOffsets(tex, texSmplr, _38, in.coord, in.compare_value.x); | ||
return out; | ||
} | ||
|
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -0,0 +1,103 @@ | ||
#pragma clang diagnostic ignored "-Wmissing-prototypes" | ||
#pragma clang diagnostic ignored "-Wmissing-braces" | ||
|
||
#include <metal_stdlib> | ||
#include <simd/simd.h> | ||
|
||
using namespace metal; | ||
|
||
template<typename T, size_t Num> | ||
struct spvUnsafeArray | ||
{ | ||
T elements[Num ? Num : 1]; | ||
|
||
thread T& operator [] (size_t pos) thread | ||
{ | ||
return elements[pos]; | ||
} | ||
constexpr const thread T& operator [] (size_t pos) const thread | ||
{ | ||
return elements[pos]; | ||
} | ||
|
||
device T& operator [] (size_t pos) device | ||
{ | ||
return elements[pos]; | ||
} | ||
constexpr const device T& operator [] (size_t pos) const device | ||
{ | ||
return elements[pos]; | ||
} | ||
|
||
constexpr const constant T& operator [] (size_t pos) const constant | ||
{ | ||
return elements[pos]; | ||
} | ||
|
||
threadgroup T& operator [] (size_t pos) threadgroup | ||
{ | ||
return elements[pos]; | ||
} | ||
constexpr const threadgroup T& operator [] (size_t pos) const threadgroup | ||
{ | ||
return elements[pos]; | ||
} | ||
}; | ||
|
||
template<typename T> struct spvRemoveReference { typedef T type; }; | ||
template<typename T> struct spvRemoveReference<thread T&> { typedef T type; }; | ||
template<typename T> struct spvRemoveReference<thread T&&> { typedef T type; }; | ||
template<typename T> inline constexpr thread T&& spvForward(thread typename spvRemoveReference<T>::type& x) | ||
{ | ||
return static_cast<thread T&&>(x); | ||
} | ||
template<typename T> inline constexpr thread T&& spvForward(thread typename spvRemoveReference<T>::type&& x) | ||
{ | ||
return static_cast<thread T&&>(x); | ||
} | ||
|
||
// Wrapper function that processes a texture gather with a constant offset array. | ||
template<typename T, template<typename, access = access::sample, typename = void> class Tex, typename Toff, typename... Tp> | ||
inline vec<T, 4> spvGatherConstOffsets(const thread Tex<T>& t, sampler s, Toff coffsets, component c, Tp... params) METAL_CONST_ARG(c) | ||
{ | ||
vec<T, 4> rslts[4]; | ||
for (uint i = 0; i < 4; i++) | ||
{ | ||
switch (c) | ||
{ | ||
case component::x: | ||
rslts[i] = t.gather(s, spvForward<Tp>(params)..., coffsets[i], component::x); | ||
break; | ||
case component::y: | ||
rslts[i] = t.gather(s, spvForward<Tp>(params)..., coffsets[i], component::y); | ||
break; | ||
case component::z: | ||
rslts[i] = t.gather(s, spvForward<Tp>(params)..., coffsets[i], component::z); | ||
break; | ||
case component::w: | ||
rslts[i] = t.gather(s, spvForward<Tp>(params)..., coffsets[i], component::w); | ||
break; | ||
} | ||
} | ||
return vec<T, 4>(rslts[0].w, rslts[1].w, rslts[2].w, rslts[3].w); | ||
} | ||
|
||
constant spvUnsafeArray<int2, 4> _30 = spvUnsafeArray<int2, 4>({ int2(-8), int2(-8, 7), int2(7, -8), int2(7) }); | ||
|
||
struct main0_out | ||
{ | ||
float4 FragColor [[color(0)]]; | ||
}; | ||
|
||
struct main0_in | ||
{ | ||
float3 coord [[user(locn0)]]; | ||
}; | ||
|
||
fragment main0_out main0(main0_in in [[stage_in]], texture2d_array<float> tex [[texture(0)]], sampler texSmplr [[sampler(0)]]) | ||
{ | ||
main0_out out = {}; | ||
out.FragColor = spvGatherConstOffsets(tex, texSmplr, _30, component::y, in.coord.xy, uint(rint(in.coord.z))); | ||
return out; | ||
} | ||
|
Oops, something went wrong.