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
|
||||
|
||||
#if defined(DATA_A_TQ2_0)
|
||||
#include "tq_utils.comp"
|
||||
#include "tq_utils.glsl"
|
||||
|
||||
vec2 dequantize(uint ib, uint iqs, uint a_offset) {
|
||||
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
|
||||
|
||||
#if defined(DATA_A_TQ1_0)
|
||||
#include "tq_utils.comp"
|
||||
#include "tq_utils.glsl"
|
||||
|
||||
vec2 dequantize(uint ib, uint iqs, uint a_offset) {
|
||||
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;
|
||||
|
||||
#include "tq_utils.comp"
|
||||
#include "tq_utils.glsl"
|
||||
|
||||
void main() {
|
||||
const uint i = gl_GlobalInvocationID.x * 4;
|
||||
|
|
|
|||
|
|
@ -2,7 +2,7 @@
|
|||
#extension GL_EXT_shader_explicit_arithmetic_types : require
|
||||
|
||||
#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;
|
||||
|
||||
|
|
|
|||
|
|
@ -113,7 +113,7 @@ FLOAT_TYPE mul_q8_1(const int32_t q_sum, const float da, const vec2 dsb, const i
|
|||
#endif
|
||||
|
||||
#if defined(DATA_A_TQ2_0)
|
||||
#include "tq_utils.comp"
|
||||
#include "tq_utils.glsl"
|
||||
|
||||
i32vec2 repack(uint ib, uint iqs) {
|
||||
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
|
||||
|
||||
#if defined(DATA_A_TQ1_0)
|
||||
#include "tq_utils.comp"
|
||||
#include "tq_utils.glsl"
|
||||
|
||||
i32vec2 repack(uint ib, uint iqs) {
|
||||
const int t00 = tq1_dequantize(ib, iqs + 0);
|
||||
|
|
|
|||
|
|
@ -1,5 +1,5 @@
|
|||
#if defined(DATA_A_TQ2_0) || defined(DATA_A_TQ1_0)
|
||||
#include "tq_utils.comp"
|
||||
#include "tq_utils.glsl"
|
||||
#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) {
|
||||
|
|
|
|||
Loading…
Reference in New Issue