Skip to content

Adding custom op in 1.14 #31349

@BMG-JTIAN

Description

@BMG-JTIAN

I noticed there was a same topic #30632. The starter of the issue switches to tf 1.12 and solves the problem. But the problem is still not resolved. When I follow the instruction and try to compile the cuda kernel, I change the cuda_kernel_helper.h to gpu_kernel_helper.h and run into the same problem.

python2.7/site-packages/tensorflow/include/tensorflow/core/util/gpu_kernel_helper.h:22:10: fatal error: third_party/gpus/cuda/include/cuda_fp16.h: No such file or directory
#include "third_party/gpus/cuda/include/cuda_fp16.h"
^~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~

Can someone figure out a reason behind this error? I pip installed tensorflow. Should I build from source?

Metadata

Metadata

Assignees

Labels

TF 1.14for issues seen with TF 1.14comp:gpuGPU related issuesstat:awaiting responseStatus - Awaiting response from authorstat:awaiting tensorflowerStatus - Awaiting response from tensorflower

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions