mirror of
https://github.com/NVIDIA/cuda-samples.git
synced 2026-06-04 00:06:52 +08:00
77 lines
2.9 KiB
Plaintext
77 lines
2.9 KiB
Plaintext
/* Copyright (c) 2026, NVIDIA CORPORATION. All rights reserved.
|
|
*
|
|
* Redistribution and use in source and binary forms, with or without
|
|
* modification, are permitted provided that the following conditions
|
|
* are met:
|
|
* * Redistributions of source code must retain the above copyright
|
|
* notice, this list of conditions and the following disclaimer.
|
|
* * Redistributions in binary form must reproduce the above copyright
|
|
* notice, this list of conditions and the following disclaimer in the
|
|
* documentation and/or other materials provided with the distribution.
|
|
* * Neither the name of NVIDIA CORPORATION nor the names of its
|
|
* contributors may be used to endorse or promote products derived
|
|
* from this software without specific prior written permission.
|
|
*
|
|
* THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS ``AS IS'' AND ANY
|
|
* EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
|
|
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
|
|
* PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR
|
|
* CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL,
|
|
* EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO,
|
|
* PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR
|
|
* PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY
|
|
* OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
* (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
* OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
*/
|
|
|
|
/*
|
|
* This CUDA Tile C++ sample demonstrates basic usage of tile
|
|
* kernels. This code launches a tile kernel using the triple chevron
|
|
* syntax and passes data between SIMT and Tile code through global
|
|
* device memory. Error checks are performed using `cudaGetLastError`
|
|
* to catch kernel launch issues and `cudaDeviceSynchronize` to catch
|
|
* kernel execution issues.
|
|
*/
|
|
|
|
#include "helper_cuda.h"
|
|
|
|
__global__ void simtKernel(int* x) {
|
|
printf("Hello, SIMT!\n");
|
|
printf("[SIMT] *x == %i\n", *x);
|
|
|
|
*x = 100;
|
|
printf("[SIMT] *x = %i\n\n", *x);
|
|
}
|
|
|
|
__tile_global__ void tileKernel(int* x) {
|
|
printf("Hello, Tile!\n");
|
|
printf("[Tile] *x == %i\n", *x);
|
|
|
|
*x = 200;
|
|
printf("[Tile] *x = %i\n\n", *x);
|
|
}
|
|
|
|
int main() {
|
|
int* d_x = nullptr;
|
|
|
|
checkCudaErrors(cudaMalloc(&d_x, sizeof(int)));
|
|
checkCudaErrors(cudaMemset(d_x, 0, sizeof(int)));
|
|
|
|
simtKernel<<<1, 1>>>(d_x);
|
|
checkCudaErrors(cudaGetLastError());
|
|
checkCudaErrors(cudaDeviceSynchronize());
|
|
|
|
/* launches tile kernel, the threads per block parameter is omitted because it must always be 1. */
|
|
tileKernel<<<1>>>(d_x);
|
|
checkCudaErrors(cudaGetLastError());
|
|
checkCudaErrors(cudaDeviceSynchronize());
|
|
|
|
int h_x = 0;
|
|
checkCudaErrors(cudaMemcpy(&h_x, d_x, sizeof(int), cudaMemcpyDeviceToHost));
|
|
checkCudaErrors(cudaFree(d_x));
|
|
|
|
printf("Hello, Host!\n");
|
|
printf("[Host] *x == %i\n", h_x);
|
|
}
|