From e4a7515dbd49bbaf11c177d13aff3986f4493dc2 Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Thu, 20 Aug 2015 18:10:35 -0500
Subject: [PATCH] More command line utility fixes
---
bin/loopy | 8 +++++---
1 file changed, 5 insertions(+), 3 deletions(-)
diff --git a/bin/loopy b/bin/loopy
index 9d667ec2f..8f365c7be 100644
--- a/bin/loopy
+++ b/bin/loopy
@@ -52,7 +52,7 @@ def main():
parser.add_argument("infile")
parser.add_argument("outfile")
- parser.add_argument("--lang")
+ parser.add_argument("--lang", metavar="LANGUAGE", help="loopy|fortran")
parser.add_argument("--target")
parser.add_argument("--name")
parser.add_argument("--transform")
@@ -75,7 +75,9 @@ def main():
lang = {
".loopy": "loopy",
".floopy": "fortran",
- }
+ ".f90": "fortran",
+ ".fpp": "fortran",
+ }.get(ext)
with open(args.infile, "r") as infile_fd:
infile_content = infile_fd.read()
@@ -130,7 +132,7 @@ def main():
kernels = [kernel]
- elif args.lang in ["fortran", "floopy", "fpp"]:
+ elif lang in ["fortran", "floopy", "fpp"]:
pre_transform_code = None
if args.transform:
with open(args.transform, "r") as xform_fd:
--
GitLab