Method dispatch_unary_vec4

cake-core/src/backends/vulkan/mod.rs:858–879

(&self, x: &Tensor, entry: &str)

    }

    fn dispatch_unary_vec4(&self, x: &Tensor, entry: &str) -> Result<Tensor> {
        let dtype = x.dtype();
        let shape = x.shape().clone();
        let n = x.elem_count();

        let buf_a = self.get_or_upload(x)?;
        let buf_out = self.alloc_output(n);

        let threads_needed = (n as u32).div_ceil(4);
        let result = self.dispatch_compute(
            entry,
            &[buf_a.buffer, self.dummy_buf.buffer, buf_out.buffer],
            &buf_out,
            n,
            &[n as u32, 0, 0, 0],
            (threads_needed.div_ceil(WG_ELEM), 1, 1),
        );

        let tensor = Self::from_f32_vec(result, shape.dims(), dtype)?;
        self.cache_activation(tensor.id(), buf_out);
        Ok(tensor)
    }

    // ── GPU softmax ──────────────────────────────────────────────────