Closed S1ro1 closed 4 days ago
This PR enables running CUDA kernels same way as python scripts are currently supported by submitting a file named train.cu.
train.cu
cc: @alexzhang13 @msaroufim let's move the discussion around this here.
TODO:
.cu
This PR enables running CUDA kernels same way as python scripts are currently supported by submitting a file named
train.cu
.