Skip to content
Unverified Commit 51a41901 authored by Kaushik Kulkarni's avatar Kaushik Kulkarni Committed by GitHub
Browse files

call_loopy: perform shape inference and infer loopy size parameters (#94)



* mypy show error codes

Co-authored-by: default avatarAndreas Kloeckner <inform@tiker.net>

* gets rid of unnecessary type-ignore

* dim_to_index_lambda_components: makes interface more convenient

caller can pass None for variable name generator, in which the routine will
create a unique name generator of its own.

* call_loopy: shape inference implementation

Co-authored-by: default avatarAndreas Kloeckner <inform@tiker.net>

* test call_loopy with shape inference

* move SizeParamGatherer to pytato.transform

* starts being less invasive about overwriting dtypes

* Add, clean up *Gatherer docs

Co-authored-by: default avatarAndreas Kloeckner <inform@tiker.net>
parent 81149098
Loading
Loading
Loading
Pipeline #179508 passed with stage
in 2 minutes and 5 seconds