Rename tq_utils.comp to tq_utils.glsl
Sponsored-by: Tether Inc. Signed-off-by: Italo Nicola <italo.nicola@collabora.com>
This commit is contained in:
parent
db52ab40fc
commit
e8133151a1
|
|
@ -422,7 +422,7 @@ vec4 dequantize4(uint ib, uint iqs, uint a_offset) {
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#if defined(DATA_A_TQ2_0)
|
#if defined(DATA_A_TQ2_0)
|
||||||
#include "tq_utils.comp"
|
#include "tq_utils.glsl"
|
||||||
|
|
||||||
vec2 dequantize(uint ib, uint iqs, uint a_offset) {
|
vec2 dequantize(uint ib, uint iqs, uint a_offset) {
|
||||||
return vec2(tq2_dequantize(ib + a_offset, iqs), tq2_dequantize(ib + a_offset, iqs + 1));
|
return vec2(tq2_dequantize(ib + a_offset, iqs), tq2_dequantize(ib + a_offset, iqs + 1));
|
||||||
|
|
@ -438,7 +438,7 @@ vec4 dequantize4(uint ib, uint iqs, uint a_offset) {
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#if defined(DATA_A_TQ1_0)
|
#if defined(DATA_A_TQ1_0)
|
||||||
#include "tq_utils.comp"
|
#include "tq_utils.glsl"
|
||||||
|
|
||||||
vec2 dequantize(uint ib, uint iqs, uint a_offset) {
|
vec2 dequantize(uint ib, uint iqs, uint a_offset) {
|
||||||
return vec2(tq1_dequantize(ib + a_offset, iqs), tq1_dequantize(ib + a_offset, iqs + 1));
|
return vec2(tq1_dequantize(ib + a_offset, iqs), tq1_dequantize(ib + a_offset, iqs + 1));
|
||||||
|
|
|
||||||
|
|
@ -13,7 +13,7 @@ layout (push_constant) uniform parameter {
|
||||||
|
|
||||||
layout (local_size_x = 256, local_size_y = 1, local_size_z = 1) in;
|
layout (local_size_x = 256, local_size_y = 1, local_size_z = 1) in;
|
||||||
|
|
||||||
#include "tq_utils.comp"
|
#include "tq_utils.glsl"
|
||||||
|
|
||||||
void main() {
|
void main() {
|
||||||
const uint i = gl_GlobalInvocationID.x * 4;
|
const uint i = gl_GlobalInvocationID.x * 4;
|
||||||
|
|
|
||||||
|
|
@ -2,7 +2,7 @@
|
||||||
#extension GL_EXT_shader_explicit_arithmetic_types : require
|
#extension GL_EXT_shader_explicit_arithmetic_types : require
|
||||||
|
|
||||||
#include "mul_mat_vec_base.glsl"
|
#include "mul_mat_vec_base.glsl"
|
||||||
#include "tq_utils.comp"
|
#include "tq_utils.glsl"
|
||||||
|
|
||||||
layout(local_size_x_id = 0, local_size_y = 1, local_size_z = 1) in;
|
layout(local_size_x_id = 0, local_size_y = 1, local_size_z = 1) in;
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -113,7 +113,7 @@ FLOAT_TYPE mul_q8_1(const int32_t q_sum, const float da, const vec2 dsb, const i
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#if defined(DATA_A_TQ2_0)
|
#if defined(DATA_A_TQ2_0)
|
||||||
#include "tq_utils.comp"
|
#include "tq_utils.glsl"
|
||||||
|
|
||||||
i32vec2 repack(uint ib, uint iqs) {
|
i32vec2 repack(uint ib, uint iqs) {
|
||||||
const int t00 = tq2_dequantize(ib, iqs + 0);
|
const int t00 = tq2_dequantize(ib, iqs + 0);
|
||||||
|
|
@ -139,7 +139,7 @@ ACC_TYPE mul_q8_1(const int32_t q_sum, const float da, const vec2 dsb, const int
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
#if defined(DATA_A_TQ1_0)
|
#if defined(DATA_A_TQ1_0)
|
||||||
#include "tq_utils.comp"
|
#include "tq_utils.glsl"
|
||||||
|
|
||||||
i32vec2 repack(uint ib, uint iqs) {
|
i32vec2 repack(uint ib, uint iqs) {
|
||||||
const int t00 = tq1_dequantize(ib, iqs + 0);
|
const int t00 = tq1_dequantize(ib, iqs + 0);
|
||||||
|
|
|
||||||
|
|
@ -1,5 +1,5 @@
|
||||||
#if defined(DATA_A_TQ2_0) || defined(DATA_A_TQ1_0)
|
#if defined(DATA_A_TQ2_0) || defined(DATA_A_TQ1_0)
|
||||||
#include "tq_utils.comp"
|
#include "tq_utils.glsl"
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
void load_a_to_shmem(const uint pos_a, const uint row, const uint col, const uint idx_m, const uint block, const uint end_k) {
|
void load_a_to_shmem(const uint pos_a, const uint row, const uint col, const uint idx_m, const uint block, const uint end_k) {
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue