You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Slides: The powerpoint file cuda_mode_lecture2.pptx can be found in the root directory of this repository. Alternatively here as Google docs presentation.
Examples: Please make sure PyTorch (2.1.2) and cuda-toolkit (nvcc compiler) are installed.
vector_addition: Classic CUDA C example, to compile use make in the vector_addition directory.
rgb_to_grayscale: Example uses PyTorch's torch.utils.cpp_extension.load_inline feature to compile a custom RGB to grayscale kernel and uses it to convert input image to grayscale and which is saved in as output.png. Run in the rgb_to_grayscale folder python rgb_to_grayscale.py.
mean_filter: This example also uses the PyTorch's cpp_extension.load_inline feature to compile a mean filter kernel. The kernel read pixel values in the surrounding (square area) of a pixel and computes the average value for each RGB channel individualy. The result is saved to output.png. Run in the mean_filter folder python mean_filter.py.