Skip to content

Commit 15a7d65

Browse files
committed
Fix incorrect type in vector_add example
1 parent 1c85eca commit 15a7d65

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

examples/vector_add/main.cu

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,9 +17,9 @@ void cuda_check(cudaError_t code) {
1717
template<int N>
1818
__global__ void my_kernel(
1919
int length,
20-
const kf::unaligned_halfX<N>* input,
20+
const kf::unaligned_vec<__half, N>* input,
2121
double constant,
22-
kf::unaligned_floatX<N>* output) {
22+
kf::unaligned_vec<float, N>* output) {
2323
int i = blockIdx.x * blockDim.x + threadIdx.x;
2424

2525
if (i * N < length) {
@@ -86,4 +86,4 @@ int main() {
8686

8787
std::cout << "result correct\n";
8888
return EXIT_SUCCESS;
89-
}
89+
}

0 commit comments

Comments
 (0)