import tensorflow.compat.v1 as tf

tf.disable_v2_behavior()

import numpy as np

import time

import sys

NUM_THREADS = int(sys.argv[1])

N = 30000

print("Num threads", NUM_THREADS)

x_data = np.random.uniform(-1, 1, N)

y_data = np.random.uniform(-1, 1, N)

x = tf.placeholder("float64", name="x")

y = tf.placeholder("float64", name="y")

hit_tests = x ** 2 + y ** 2 < 1.0

hits = tf.reduce_sum(tf.cast(hit_tests, "int32"))

with tf.Session(

config=tf.ConfigProto(

inter_op_parallelism_threads=NUM_THREADS,

intra_op_parallelism_threads=NUM_THREADS,

)

) as sess:

start = time.time()

for i in range(10000):

sess.run(hits, {x: x_data, y: y_data})

print(time.time() - start)