-
Notifications
You must be signed in to change notification settings - Fork 617
Closed
Labels
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-gpuas the docker image - Specify the
TF_NEED_CUDAvariable appropriately in our scripts
facaiy and ramanshrivastava