add kernel with torch

train/run_cuda.py

@@ -0,0 +1,14 @@
+import torch
+from torch.utils.cpp_extension import load
+
+# Load the compiled CUDA extension
+my_cuda_extension = load(name='my_cuda_kernel', sources=['my_cuda_kernel.cu'])
+
+# Define a PyTorch tensor on the GPU
+input_tensor = torch.cuda.FloatTensor([1, 2, 3, 4])
+
+# Call the CUDA function
+output_tensor = my_cuda_extension.my_cuda_function(input_tensor)
+
+# Print the result
+print(output_tensor)