fix j-major wrong ne on CDNA
This commit is contained in:
parent
6acad9c759
commit
cffa070b08
|
|
@ -446,7 +446,7 @@ namespace ggml_cuda_mma {
|
||||||
static constexpr int J = J_;
|
static constexpr int J = J_;
|
||||||
static constexpr data_layout dl = DATA_LAYOUT_J_MAJOR;
|
static constexpr data_layout dl = DATA_LAYOUT_J_MAJOR;
|
||||||
|
|
||||||
static constexpr int ne = I * J / 32;
|
static constexpr int ne = tile<I_, J_, T, DATA_LAYOUT_I_MAJOR>::ne;
|
||||||
T x[ne] = {0};
|
T x[ne] = {0};
|
||||||
|
|
||||||
static constexpr __device__ bool supported() {
|
static constexpr __device__ bool supported() {
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue