def get_context(args):
context = None<EOL>if args.gpu is None or args.gpu == '<STR_LIT>':<EOL><INDENT>context = mx.cpu()<EOL><DEDENT>if isinstance(args.gpu, int):<EOL><INDENT>context = mx.gpu(args.gpu)<EOL><DEDENT>return context<EOL>
This method gets context of execution