Skip to content

euclidean_distance_transform_op.cc build falied on GPU #349

@facaiy

Description

@facaiy

System information

  • Have I written custom code (as opposed to using a stock example script provided in TensorFlow): No
  • OS Platform and Distribution (e.g., Linux Ubuntu 16.04): Linux
  • TensorFlow installed from (source or binary): binary
  • TensorFlow version (use command below): tf-nightly-gpu-2.0-preview
  • TensorFlow Addons installed from (source, PyPi): source
  • TensorFlow Addons version: master
  • Python version and type (eg. Anaconda Python, Stock Python as in Mac, or homebrew installed Python etc): tensorflow/tensorflow:custom-op-gpu
  • Bazel version (if compiling from source): N/A
  • GCC/Compiler version (if compiling from source):
  • Is GPU used? (yes/no): yes
  • GPU model (if used):

Describe the bug

When I'm working in #294, tensorflow_addons/custom_ops/image/cc/kernels/euclidean_distance_transform_op.cc won't compile with gpu successfully. So I disable its gpu kernel registration temporality, and hope it will be fixed later.

Describe the expected behavior

Code to reproduce the issue

Other info / logs

Metadata

Metadata

Assignees

Labels

bugSomething isn't workingimage

Type

No type

Projects

No projects

Milestone

No milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions