File tree Expand file tree Collapse file tree 2 files changed +2
-1
lines changed
Expand file tree Collapse file tree 2 files changed +2
-1
lines changed Original file line number Diff line number Diff line change @@ -25,6 +25,7 @@ static bool ggml_is_view(const struct ggml_tensor * t) {
2525// ops that return true for this function must not use restrict pointers for their backend implementations
2626bool ggml_op_can_inplace (enum ggml_op op ) {
2727 switch (op ) {
28+ case GGML_OP_FILL :
2829 case GGML_OP_SCALE :
2930 case GGML_OP_DIAG_MASK_ZERO :
3031 case GGML_OP_DIAG_MASK_INF :
Original file line number Diff line number Diff line change 44#define CUDA_FILL_BLOCK_SIZE 256
55
66template <typename T>
7- static __global__ void fill_kernel (T * __restrict__ dst, const int64_t k, const T value) {
7+ static __global__ void fill_kernel (T * dst, const int64_t k, const T value) {
88 const int64_t i = (int64_t )blockDim .x * blockIdx .x + threadIdx .x ;
99 if (i >= k) {
1010 return ;
You can’t perform that action at this time.
0 commit comments