소스 검색

CUDA: non-contiguous src0 not supported for PAD (#15869)

Sigbjørn Skjæret 4 달 전
부모
커밋
5ef22d281d
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  1. 1 1
      ggml/src/ggml-cuda/ggml-cuda.cu

+ 1 - 1
ggml/src/ggml-cuda/ggml-cuda.cu

@@ -3574,9 +3574,9 @@ static bool ggml_backend_cuda_device_supports_op(ggml_backend_dev_t dev, const g
         case GGML_OP_SUM_ROWS:
         case GGML_OP_MEAN:
         case GGML_OP_GROUP_NORM:
+        case GGML_OP_PAD:
             return ggml_is_contiguous(op->src[0]);
         case GGML_OP_UPSCALE:
-        case GGML_OP_PAD:
         case GGML_OP_PAD_REFLECT_1D:
         case GGML_OP_ARANGE:
         case GGML_OP_TIMESTEP_EMBEDDING: