Skip to content

Commit 25e6a70

Browse files
more cleanup
1 parent c9f4400 commit 25e6a70

File tree

2 files changed

+0
-19
lines changed

2 files changed

+0
-19
lines changed

csrc/kernels.cu

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -2025,12 +2025,6 @@ __global__ void kspmm_coo_very_sparse_naive(
20252025
}
20262026
}
20272027

2028-
template <typename T> __device__ void printnonzero(T* A, int num_values, const char* strval) {
2029-
for (int i = 0; i < num_values; i++)
2030-
if ((float)A[i] != 0.0)
2031-
printf("%s %i %f\n", strval, i, (float)A[i]);
2032-
}
2033-
20342028
#define num_values_4bit 32
20352029

20362030
template <typename T, int THREADS, int BITS>
@@ -2508,6 +2502,3 @@ MAKE_OptimizerStatic8bit1StateBlockwise(LION, __nv_bfloat16, 256, 1)
25082502
MAKE_OptimizerStatic8bit1StateBlockwise(ADAGRAD, float, 256, 1)
25092503
MAKE_OptimizerStatic8bit1StateBlockwise(ADAGRAD, half, 256, 1)
25102504
MAKE_OptimizerStatic8bit1StateBlockwise(ADAGRAD, __nv_bfloat16, 256, 1)
2511-
2512-
template __device__ void printnonzero<float>(float* A, int num_values, const char* strval);
2513-
template __device__ void printnonzero<half>(half* A, int num_values, const char* strval);

csrc/kernels.hip

Lines changed: 0 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -2162,13 +2162,6 @@ __global__ void kspmm_coo_very_sparse_naive(int *max_count, int *max_idx, int *o
21622162
}
21632163
}
21642164

2165-
template <typename T> __device__ void printnonzero(T *A, int num_values, const char * strval)
2166-
{
2167-
for(int i = 0; i < num_values; i++)
2168-
if((float)A[i] != 0.0)
2169-
printf("%s %i %f\n", strval, i, (float)A[i]);
2170-
}
2171-
21722165
// No of 4bit values processed by each thread
21732166
#define num_values_4bit 32
21742167
template <typename T, int THREADS, int BITS> __global__ void kgemm_4bit_inference_naive(int M, int N, int K, T * __restrict__ const A, unsigned char *B, float *absmax, const float *datatype, T * out, int lda, int ldb, int ldc, int blocksize)
@@ -2621,6 +2614,3 @@ MAKE_OptimizerStatic8bit1StateBlockwise(LION, hip_bfloat16, 256, 1)
26212614
MAKE_OptimizerStatic8bit1StateBlockwise(ADAGRAD, float, 256, 1)
26222615
MAKE_OptimizerStatic8bit1StateBlockwise(ADAGRAD, half, 256, 1)
26232616
MAKE_OptimizerStatic8bit1StateBlockwise(ADAGRAD, hip_bfloat16, 256, 1)
2624-
2625-
template __device__ void printnonzero<float>(float *A, int num_values, const char*strval);
2626-
template __device__ void printnonzero<half>(half *A, int num_values, const char*strval);

0 commit comments

Comments
 (0)