Closed
Description
Currently we've omitted the gpu kernels from addons while we got everything setup. Now that CI testing is setup for GPU we should begin adding these to addons. There's a lot of things required so I wanted to start this issue so we can track / discuss.
Some of the TODO:
- Add gpu kernels and compile them as shown in the custom_op template which now includes a GPU op : PR set up GPU environment #294
- Configure BUILD scripts to use cuda tests
- image/image_projective_transform_op_gpu.cu.cc set up GPU environment #294
- distance_transform_ops
- image/adjust_hsv_in_yiq_op_gpu.cu.cc
- seq2seq/beam_search_ops_gpu.cu.cc
- Update CONTRIBUTING documentation and add
tensorflow/tensorflow:custom-op-gpu
as the docker image - Specify the
TF_NEED_CUDA
variable appropriately in our scripts