Как сбросить тензор потока XLA LLVM IR?

Question

Я использовал следующую команду в Tensorflow 1.2:

export TF_XLA_FLAGS='--dump_ir_before_passes=true --dump_temp_products_to=./tmp'

для сброса LLVM IR в Tensorflow. Однако файл определения этого флага link_to_the_flag_definition удален в Tensorflow 1.3, и теперь мне интересно, как я могу получить дамп IR LLVM?

Вот тестовый файл для вашего удобства:

import tensorflow as tf
import numpy as np
import os
from tensorflow.python.client import timeline
import json

run_metadata = tf.RunMetadata()
sess = tf.Session()
jit_scope = tf.contrib.compiler.jit.experimental_jit_scope
x = tf.placeholder(np.float32, shape=[1000000])
y = tf.placeholder(np.float32, shape=[1000000])
c = tf.constant(0.1)

with jit_scope():
  z = tf.add(tf.scalar_mul(0.1,x), y)
  ix = np.ones((1000000), dtype=np.float32)
  iy = np.ones((1000000), dtype=np.float32)
  sess.run(z,
           feed_dict={x: ix, y: iy},
           options=tf.RunOptions(trace_level=tf.RunOptions.FULL_TRACE),
           run_metadata=run_metadata)
  trace = timeline.Timeline(step_stats=run_metadata.step_stats)
  with open('timeline.ctf.json', 'w') as trace_file:
    trace_file.write(trace.generate_chrome_trace_format())

Answer 1

Я нашел флаг --xla_dump_ir_to здесь: https://github.com/tensorflow/tensorflow/issues/11462. Это было добавлено в Tensorflow 1.3.