Skip to content
Snippets Groups Projects

Repository graph

You can move around the graph by using the arrow keys.
Select Git revision
  • cupyactx
  • main default protected
2 results
Created with Raphaël 2.2.011Jun109820May122119Apr7517Mar163115Feb23Jan1613121127Dec26241625Nov241776229Oct272422201918171511524Sep2220430Aug29262416121121Jul2017161476130Jun2928272524222120181716151413121110111098742129May2827262221205Add tag_axesMake typing in ArrayContext more precise using TypeVarsport deprecated uses of freeze and thawAdd comment to warn against early exits from freezeUse NameHint/PrefixNamed to generate better kernel names in pytato freezeArray identifiers in pytato freezes: use "_ary", not "_actx"Simplification: do not exit early out of freeze for 0-size arraysCI: modernize the freezes on DOFArraysto_numpy: thaw(freeze(...)) firstrestructure the way 'freeze' is implemented. Move the major bits to `ArrayContext.freeze`Rewrite actx.NameHint to pytato.PrefixNamed when taggingEinsum array argument naming: Use NameHintAdd NameHint tagPass name of function being compiled to pt.generate_{jax,loopy}Log function name in passes of actx.compileAdd folds in arraycontext.impl.pytatoDeprecate passing unevaluated arrays to compiled functionsDrop xfail in test_flatten_array_containerFix Jax installation on Gitlab CIStop setting PY_EXE in Gitlab CI configGeneralize Pytato Array Context to allow multiple targetstest JAXArrayContextCI: adds jax to requirementsdefines JAXArrayContextclean up downstream testsgitlab-ci: remove EXTRA_INSTALL exportsfurther clarifies the role of ArrayContext.aray_typestests to_numpy(...) on frozen arraysadds cla.Array to PytatoPyOpenCLArrayContext.array_typesimplement _with_new_tags for taggable.Axissupport unions in dataclass_array_containerwith_container_arithmetic: Default cls_has_array_context_attr based on presence of attributeadd typing to ArrayContextallow dataclass containers with only DeviceArraystransform DAG before making it a placeholderStringify return type of flat_size_and_dtype to avoid py3.9+ type subscriptAdd flat_size_and_dtypefix TaggableCLArray copy with queueMake get_container_context_recursively return actx or errorPyOpenCLArrayContext.make_einsum: use `tagged` tags for the output array as well