ocFile.close()
count *= 4
print("AutoGemm.py: %u kernels will be pre-compiled." % count)
################################################################################
# Main
################################################################################
if __name__ == "__main__":
# parse arguments
ap = argparse.ArgumentParser(description="Which gemm kernels to compile offline.")
ap.add_argument("--output-path", dest="output" )
ap.add_argument("--precisions", dest="precisions", action="store", nargs="+", choices=AutoGemmParameters.precisions )
ap.add_argument("--orders", dest="orders", action="store", nargs="+", choices=AutoGemmParameters.orders )
ap.add_argument("--transposes", dest="transposes", action="store", nargs="+", choices=AutoGemmParameters.getTransposeChoices() )
ap.add_argument("--betas", dest="betas", action="store", nargs="+", type=int, choices=AutoGemmParameters.betas )
args = ap.parse_args()
if args.output:
Common.setOutputPath(args.output)
else:
print("Warning: No output path specified; default is working directory.")
# write offline compilation header
if args.precisions is None:
args.precisions = []
if args.transposes is None:
args.transposes = []
if args.orders is None:
args.orders = []
if args.betas is None:
ap.add_argument("--output-path", dest="output")
ap.add_argument("--precisions",
dest="precisions",
action="store",
nargs="+",
choices=AutoGemmParameters.precisions)
ap.add_argument("--orders",
dest="orders",
action="store",
nargs="+",
choices=AutoGemmParameters.orders)
ap.add_argument("--transposes",
dest="transposes",
action="store",
nargs="+",
choices=AutoGemmParameters.getTransposeChoices())
ap.add_argument("--betas",
dest="betas",
action="store",
nargs="+",
type=int,
choices=AutoGemmParameters.betas)
args = ap.parse_args()
if args.output:
Common.setOutputPath(args.output)
else:
print(
"Warning: No output path specified; default is working directory.")
# write offline compilation header
if args.precisions is None:
ocFile.close()
count *= 4
print "AutoGemm.py: %u kernels will be pre-compiled." % count
################################################################################
# Main
################################################################################
if __name__ == "__main__":
# parse arguments
ap = argparse.ArgumentParser(description="Which gemm kernels to compile offline.")
ap.add_argument("--output-path", dest="output" )
ap.add_argument("--precisions", dest="precisions", action="store", nargs="+", choices=AutoGemmParameters.precisions )
ap.add_argument("--orders", dest="orders", action="store", nargs="+", choices=AutoGemmParameters.orders )
ap.add_argument("--transposes", dest="transposes", action="store", nargs="+", choices=AutoGemmParameters.getTransposeChoices() )
ap.add_argument("--betas", dest="betas", action="store", nargs="+", type=int, choices=AutoGemmParameters.betas )
args = ap.parse_args()
if args.output:
Common.setOutputPath(args.output)
else:
print "Warning: No output path specified; default is working directory."
# write offline compilation header
if args.precisions is None:
args.precisions = []
if args.transposes is None:
args.transposes = []
if args.orders is None:
args.orders = []
if args.betas is None:
