From 0f48ae92009f683837d94bfd3de2426393268638 Mon Sep 17 00:00:00 2001
From: Taylor Childers <taylor@jtchilders.com>
Date: Wed, 4 Dec 2019 15:52:13 +0000
Subject: [PATCH] update

---
 square_images/model_run.py | 68 --------------------------------------
 1 file changed, 68 deletions(-)
 delete mode 100644 square_images/model_run.py

diff --git a/square_images/model_run.py b/square_images/model_run.py
deleted file mode 100644
index 0ac43f5..0000000
--- a/square_images/model_run.py
+++ /dev/null
@@ -1,68 +0,0 @@
-import time
-
-def run(point):
-   start = time.time()
-   try:
-      batch_size = point['batch_size']
-      image_size = point['image_size']
-      in_channels = point['in_channels']
-      out_channels = point['out_channels']
-      kernel_size = point['kernel_size']
-      omp_num_threads = point['omp_num_threads']
-
-      import os
-      os.environ['OMP_NUM_THREADS'] = str(omp_num_threads)
-      os.environ['MKL_NUM_THREADS'] = str(omp_num_threads)
-      os.environ['KMP_HW_SUBSET'] = '1s,%sc,2t' % str(omp_num_threads)
-      os.environ['KMP_AFFINITY'] = 'granularity=fine,verbose,compact,1,0'
-      os.environ['KMP_BLOCKTIME'] = str(0)
-      #os.environ['MKLDNN_VERBOSE'] = str(1)
-      import torch
-
-      print('torch version: ',torch.__version__,' torch file: ',torch.__file__)
-
-
-      inputs = torch.arange(batch_size * image_size * image_size * in_channels,dtype=torch.float).view((batch_size,in_channels,image_size,image_size))
-      
-      layer = torch.nn.Conv2d(in_channels,out_channels,(kernel_size,kernel_size),stride=1)
-      outputs = layer(inputs)
-
-      total_flop = kernel_size * kernel_size * in_channels * out_channels * outputs.shape[-1] * outputs.shape[-2] * batch_size
-      
-      runs = 5
-      tot_time = 0.
-      tt = time.time()
-      for _ in range(runs):
-         outputs = layer(inputs)
-         tot_time += time.time() - tt
-         tt = time.time()
-
-      ave_time = tot_time / runs
-
-      print('total_flop = ',total_flop,'ave_time = ',ave_time)
-
-      ave_flops = total_flop / ave_time
-      runtime = time.time() - start
-      print('runtime=',runtime,'ave_flops=',ave_flops)
-
-      return ave_flops
-   except Exception as e:
-      import traceback
-      print('received exception: ',str(e))
-      print(traceback.print_exc())
-      print('runtime=',time.time() - start)
-      return 0.
-
-
-if __name__ == '__main__':
-   point = {
-      'batch_size': 10,
-      'image_size': 512,
-      'in_channels': 3,
-      'out_channels': 64,
-      'kernel_size': 4,
-      'omp_num_threads':64,
-   }
-
-   print('flops for this setting =',run(point))
-