|
@@ -69,7 +69,6 @@ void ggml_cuda_op_im2col(ggml_backend_cuda_context & ctx, ggml_tensor * dst) {
|
|
|
float * dst_d = (float *)dst->data;
|
|
float * dst_d = (float *)dst->data;
|
|
|
cudaStream_t stream = ctx.stream();
|
|
cudaStream_t stream = ctx.stream();
|
|
|
|
|
|
|
|
- GGML_ASSERT(src0->type == GGML_TYPE_F16);
|
|
|
|
|
GGML_ASSERT(src1->type == GGML_TYPE_F32);
|
|
GGML_ASSERT(src1->type == GGML_TYPE_F32);
|
|
|
GGML_ASSERT( dst->type == GGML_TYPE_F16 || dst->type == GGML_TYPE_F32);
|
|
GGML_ASSERT( dst->type == GGML_TYPE_F16 || dst->type == GGML_TYPE_F32);
|
|
|
|
|
|