Spaces:
Running
Running
Commit
·
91954a7
1
Parent(s):
ba60f98
CUDA: remove bad assert (ggml/972)
Browse files
ggml/src/ggml-cuda/im2col.cu
CHANGED
|
@@ -69,7 +69,6 @@ void ggml_cuda_op_im2col(ggml_backend_cuda_context & ctx, ggml_tensor * dst) {
|
|
| 69 |
float * dst_d = (float *)dst->data;
|
| 70 |
cudaStream_t stream = ctx.stream();
|
| 71 |
|
| 72 |
-
GGML_ASSERT(src0->type == GGML_TYPE_F16);
|
| 73 |
GGML_ASSERT(src1->type == GGML_TYPE_F32);
|
| 74 |
GGML_ASSERT( dst->type == GGML_TYPE_F16 || dst->type == GGML_TYPE_F32);
|
| 75 |
|
|
|
|
| 69 |
float * dst_d = (float *)dst->data;
|
| 70 |
cudaStream_t stream = ctx.stream();
|
| 71 |
|
|
|
|
| 72 |
GGML_ASSERT(src1->type == GGML_TYPE_F32);
|
| 73 |
GGML_ASSERT( dst->type == GGML_TYPE_F16 || dst->type == GGML_TYPE_F32);
|
| 74 |
|