feat(example): impl. vadd

ze/kernel.go

@@ -0,0 +1,41 @@
+package ze
+
+import (
+	"runtime"
+	"unsafe"
+
+	"github.com/fumiama/gozel"
+)
+
+// KernelHandle is a handle to a Level Zero kernel.
+type KernelHandle gozel.ZeKernelHandle
+
+// KernelCreate creates a kernel from the module by the given function name.
+func (h ModuleHandle) KernelCreate(kernelName string) (KernelHandle, error) {
+	b := []byte(kernelName + "\x00")
+	var k gozel.ZeKernelHandle
+	_, err := gozel.ZeKernelCreate(gozel.ZeModuleHandle(h), &gozel.ZeKernelDesc{
+		Stype:       gozel.ZE_STRUCTURE_TYPE_KERNEL_DESC,
+		Pkernelname: &b[0],
+	}, &k)
+	runtime.KeepAlive(b)
+	return KernelHandle(k), err
+}
+
+// SetArgumentValue sets the value of a kernel argument at the given index.
+func (h KernelHandle) SetArgumentValue(argIndex uint32, argSize uintptr, pArgValue unsafe.Pointer) error {
+	_, err := gozel.ZeKernelSetArgumentValue(gozel.ZeKernelHandle(h), argIndex, argSize, pArgValue)
+	return err
+}
+
+// SetGroupSize sets the thread group size for the kernel.
+func (h KernelHandle) SetGroupSize(groupSizeX uint32, groupSizeY uint32, groupSizeZ uint32) error {
+	_, err := gozel.ZeKernelSetGroupSize(gozel.ZeKernelHandle(h), groupSizeX, groupSizeY, groupSizeZ)
+	return err
+}
+
+// Destroy destroys the kernel and releases its resources.
+func (h KernelHandle) Destroy() error {
+	_, err := gozel.ZeKernelDestroy(gozel.ZeKernelHandle(h))
+	return err
+}