"Comptime" accelerates Python code by precomputing complex calculations, turning them into simple lookups for faster execution.
Project description
comptime
"Comptime" accelerates Python code by precomputing complex calculations, turning them into simple lookups for faster execution.
Table of Contents
Comptime is inspired by the concept of compile-time computation found in languages like Zig, and it brings this powerful feature into the world of Python, an interpreted language. By utilizing special decorators, you can mark functions whose return values should be precomputed. This enables you to separate computationally expensive parts of your code, performing those calculations once, and embedding the results directly into your source code. The result is Python code that executes faster by turning complex calculations into simple lookups. Whether you're optimizing critical performance bottlenecks or exploring new ways to structure your code, Comptime offers a novel approach to accelerate your Python development.
"Zig into Python's speed lane with Comptime – it's not a sprint; it's a compile!"
:warning: Warning: this is only a proof-of-concept
This code was only created to test the concept. It should NOT be used in a production environment, as I can not guarantee at this time that the semantics of the outputted code are correct. Furthermore, initial performance tests show that for relatively simple calculations, this does NOT actually improve performance:
python examples/perf.py # pre-comptime:
# Function executed in: 2.0992 seconds total; avg of 209.92 ns. per execution.
python examples/perf_match.py # comptime with match-case strategy:
# Function executed in: 21.5824 seconds total; avg of 2158.24 ns. per execution.
python examples/perf_dict.py # comptime with dict lookup strategy:
# Function executed in: 152.4098 seconds total; avg of 15240.98 ns. per execution.
While this package could be useful in cases where the calculation is actually heavy, it could also negatively impact your performance!
Installation
pip install comptime
Usage
# src_raw/main.py: before comptime
from comptime import comptime
@comptime.skip
def nonpure_method():
# this method has side effects so should only be used at runtime
print("e.g. sending an email")
@comptime
def my_method():
...
# some expensive calculations
nonpure_method() # not executed due to @comptime.skip
return 41 + 1
@comptime("users", "posts")
def call_api(endpoint):
# api get value for endpoint
...
return value
call_api("other") # possible comptime warning due to unsupported argument?
comptime --input src_raw --output src_compiled
# src_compiled/main.py: after comptime
import typing
def nonpure_method():
# this method has side effects so should only be used at runtime
print("e.g. sending an email")
def my_method():
# computed by comptime
return 42
def call_api(endpoint: typing.Literal["users", "posts"]):
# computed by comptime
match endpoint:
case "users":
return ['user1']
case "posts":
return ['post1']
case _:
raise ValueError("Uncompiled variant {endpoint}")
return value
Acknowledgments
This project owes its inspiration and certain elements of its design to various sources:
-
Zig Programming Language: The concept of compile-time computation in Comptime is inspired by Zig, a language that emphasizes safety, performance, and readability. A special thank you to Zig's creators and community for their innovative approach to programming.
-
GPT-4 by OpenAI: Assistance with the project boilerplate, creative brainstorming, and crafting the memorable slogan "Zig into Python's speed lane with Comptime – it's not a sprint; it's a compile!" was provided by GPT-4. It also wrote this section.
Please note that the author of comptime is not affiliated with Zig, OpenAI, or any other entities mentioned above. The acknowledgments are expressions of gratitude and inspiration and do not imply any formal association or endorsement by these parties.
License
comptime
is distributed under the terms of the MIT license.
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.