From 8d7097a03db9700ad54b8793563be1422b62e778 Mon Sep 17 00:00:00 2001 From: x00373192 Date: Tue, 27 Apr 2021 13:13:14 +0800 Subject: [PATCH] DTS:DTS202103090GHWDGP0E00 Description: ascend adapter for tensorflow 2.x Team:HISI_DRV Feature or Bugfix:Feature --- tf_adapter_2.x/examples/demo.py | 41 +++++++++++++++++++++++++++++++++ 1 file changed, 41 insertions(+) create mode 100755 tf_adapter_2.x/examples/demo.py diff --git a/tf_adapter_2.x/examples/demo.py b/tf_adapter_2.x/examples/demo.py new file mode 100755 index 000000000..e5cae4ab8 --- /dev/null +++ b/tf_adapter_2.x/examples/demo.py @@ -0,0 +1,41 @@ +import os + +os.environ['NPU_ENABLE_PERF'] = 'true' +os.environ['NPU_DUMP_GRAPH'] = 'true' +os.environ['DUMP_GE_GRAPH'] = '2' +os.environ['DUMP_GRAPH_LEVEL'] = '1' + +from absl import logging + +logging.set_verbosity(logging.INFO) +logging.info("123") + +# import tensorflow as tf +# import npu_device as npu +# +# npu.open().as_default() +# +# +# @npu.never_nested_function +# def aic_floating_poit_overflow(): +# float_status = npu.gen_npu_ops.npu_alloc_float_status() +# assign_float_status = npu.gen_npu_ops.npu_get_float_status(float_status) +# cleared_float_status = npu.gen_npu_ops.npu_clear_float_status(assign_float_status) +# return tf.reduce_all(tf.equal(float_status, cleared_float_status)) +# +# +# @tf.function +# def f(x): +# y = tf.add(3.4e+38, x) +# with tf.control_dependencies([y]): +# return aic_floating_poit_overflow(), aic_floating_poit_overflow() +# +# +# print(f(tf.constant(0.0, dtype=tf.float32))) +# print(f(tf.constant(100.0, dtype=tf.float32))) +# print(f(tf.constant(3.4e+38, dtype=tf.float32))) +# print(f(tf.constant(100.0, dtype=tf.float32))) + + + + -- Gitee