Check dtypes in `fnsm.match_t_unit_to_einsum` about feinsum HOT 1 CLOSED

kaushikcfd commented on July 17, 2024

Check dtypes in `fnsm.match_t_unit_to_einsum`

from feinsum.

Comments (1)

kaushikcfd commented on July 17, 2024

Done, see

feinsum/src/feinsum/loopy_utils.py

Lines 107 to 114 in fbd94ff

 def _get_dtype_for_subst_argument(t_unit: lp.TranslationUnit, 

 kernel_name: str, 

 rule_name: str) -> np.dtype[Any]: 

 """ 

  Returns the inferred data type for *rule_name*'s expression when all the 

  arguments passed to it are equal to it kernel's 

  :attr:`loopy.LoopKernel.index_dtype`. 

  """

from feinsum.

Related Issues (20)

Add a utility to check whether two Fused-einsums are performance-isomorphic. HOT 1
Avoid `lp.fix_parameters` while timing/empirically verifying the transformation
Fusion algorithm for composition of batched einsums
Add .rtd-env-py3.yml
Add empirical global memory bandwidth measurement
Add empirical maximum FLOP rate measurement HOT 2
Add helpers to compare the best result in the database with other tools HOT 1
Add a CI to do some trivial checks on the database entries
Remove dependency on OpenTuner. HOT 4
Add parallel tuning capability HOT 5
Use vertex coloring while canonicalizing the FusedEinsum HOT 1
Support matching `max` and `min` reductions
Rename FusedEinsum -> BatchedEinsum
[docs] clarify that we use the same namespace for argument names and index names
Pretty print a Batched einsum
Validate the prescription of the transform space by fuzz testing
feinsum.query should accept a cl.Device instead of cl.Context
Check statement dependencies in `f.match_t_unit_to_einsum`
Database should not store the kernel string rather a filepath that can be located HOT 1
`f.loopy_utils` should provide a transformation to make i-th einsum operand as substitution

Check dtypes in `fnsm.match_t_unit_to_einsum` about feinsum HOT 1 CLOSED

Comments (1)

Related Issues (20)

Recommend Projects

React

Vue.js

Typescript

TensorFlow

Django

Laravel

D3

Recommend Topics

javascript

web

server

Machine learning

Visualization

Game

Recommend Org

Facebook

Microsoft

Google

Alibaba

D3

Tencent

	def _get_dtype_for_subst_argument(t_unit: lp.TranslationUnit,
	kernel_name: str,
	rule_name: str) -> np.dtype[Any]:
	"""
	Returns the inferred data type for rule_name's expression when all the
	arguments passed to it are equal to it kernel's
	:attr:`loopy.LoopKernel.index_dtype`.
	"""