def setUp(self):
log_file = "pytorch_conv2D_winograd.log"
is_debug = True
set_up_logging(log_file=log_file, is_debug=is_debug)
self.logger = get_logger(name=__name__)
self.logger.setLevel(logging.DEBUG)
self.logger.info("Set up test")
def setUp(self):
log_file = "conv2D_benchmark.log"
is_debug = True
set_up_logging(log_file=log_file, is_debug=is_debug)
self.logger = get_logger(name=__name__)
self.logger.setLevel(logging.DEBUG)
self.logger.info("Set up test")
self.dtype = torch.float
if torch.cuda.is_available():
self.device = torch.device("cuda")
else:
self.device = torch.device("cpu")
self.conv_exec_type = ConvExecType.SGEMM
def setUp(self):
log_file = "pytorch_conv2D_reuse_map_fft.log"
is_debug = True
set_up_logging(log_file=log_file, is_debug=is_debug)
self.logger = get_logger(name=__name__)
self.logger.setLevel(logging.DEBUG)
self.logger.info("Set up test")
if torch.cuda.is_available():
self.device = torch.device('cuda')
print("cuda is available")
else:
self.device = torch.device('cpu')
print("no cuda device is available")
self.dtype = torch.float
self.ERR_MESSAGE_ALL_CLOSE = "The expected array desired and " \
"computed actual are not almost equal."