Really appreciate this great work, it help me build tf_custom_op_library successfully. Furthermore, I want to add GPU kernel for my tf_custom_op_library, how to integrate gpu files with
|
def tf_custom_op_library( |
?
I have tried to use load("@org_tensorflow//tensorflow:tensorflow.bzl", "tf_custom_op_library") from TF sourcecode directly, but it gives me error like this:
ERROR: /workspaces/cache/bazel/external/org_tensorflow/tensorflow/BUILD:1153:21: declared output 'external/org_tensorflow/tensorflow/libtensorflow_framework.so.2' is a dangling symbolic link
ERROR: /workspaces/cache/bazel/external/org_tensorflow/tensorflow/BUILD:1153:21: Executing genrule @org_tensorflow//tensorflow:libtensorflow_framework.so.2_sym failed: not all outputs were created or valid
Really appreciate this great work, it help me build tf_custom_op_library successfully. Furthermore, I want to add GPU kernel for my tf_custom_op_library, how to integrate gpu files with
federated-compute/fcp/tensorflow/pip_tf/defs.bzl
Line 96 in 616b1d0
I have tried to use
load("@org_tensorflow//tensorflow:tensorflow.bzl", "tf_custom_op_library")from TF sourcecode directly, but it gives me error like this: