Add pt.(zeros|ones)_like

[kaushikcfd]

No description provided.

[kaushikcfd]

Hi @inducer,
So the leftover warnings here are:

/home/runner/work/pytato/pytato/test/test_codegen.py:2092:21 - warning: Type of parameter "ctx_factory" is unknown (reportUnknownParameterType)
Warning: Type of parameter "ctx_factory" is unknown (reportUnknownParameterType)
/home/runner/work/pytato/pytato/test/test_codegen.py:2092:21 - warning: Type annotation is missing for parameter "ctx_factory" (reportMissingParameterType)
Warning: Type annotation is missing for parameter "ctx_factory" (reportMissingParameterType)
/home/runner/work/pytato/pytato/test/test_codegen.py:2096:5 - warning: Type of "ctx" is unknown (reportUnknownVariableType)
Warning: Type of "ctx" is unknown (reportUnknownVariableType)
/home/runner/work/pytato/pytato/test/test_codegen.py:2097:5 - warning: Type of "cq" is unknown (reportUnknownVariableType)
Warning: Type of "cq" is unknown (reportUnknownVariableType)
/home/runner/work/pytato/pytato/test/test_codegen.py:2097:10 - warning: Type of "CommandQueue" is unknown (reportUnknownMemberType)
Warning: Type of "CommandQueue" is unknown (reportUnknownMemberType)
/home/runner/work/pytato/pytato/test/test_codegen.py:2106:11 - warning: Type of "generate_loopy" is partially unknown
  Type of "generate_loopy" is "(result: Array | DictOfNamedArrays | dict[str, Array], target: LoopyTarget | None = None, options: Options | None = None, *, function_name: str = "_pt_kernel", cl_device: Unknown | None = None, array_tag_t_to_not_propagate: frozenset[type[Tag]] = frozenset([ImplStored, Named, PrefixNamed]), axis_tag_t_to_not_propagate: frozenset[type[Tag]] = frozenset()) -> BoundProgram" (reportUnknownMemberType)
Warning: Type of "generate_loopy" is partially unknown
  Type of "generate_loopy" is "(result: Array | DictOfNamedArrays | dict[str, Array], target: LoopyTarget | None = None, options: Options | None = None, *, function_name: str = "_pt_kernel", cl_device: Unknown | None = None, array_tag_t_to_not_propagate: frozenset[type[Tag]] = frozenset([ImplStored, Named, PrefixNamed]), axis_tag_t_to_not_propagate: frozenset[type[Tag]] = frozenset()) -> BoundProgram" (reportUnknownMemberType)
/home/runner/work/pytato/pytato/test/test_codegen.py:2107:5 - warning: Type of "_" is Any (reportAny)
Warning: Type of "_" is Any (reportAny)
/home/runner/work/pytato/pytato/test/test_codegen.py:2107:8 - warning: Type of "pt_out" is Any (reportAny)
Warning: Type of "pt_out" is Any (reportAny)
Warning: Argument type is Any
  Argument corresponds to parameter "actual" in function "assert_allclose" (reportAny)

0 errors, 10 warnings, 0 notes

Any preferences on how many of them we should fix and how should we skip?

Thanks

[inducer]

See latest commit, I just disabled those warnings for the tests. Some of them may disappear if/when pyopencl gets typed.

[kaushikcfd]

Thanks!

[inducer]

We should add some discussion of this in design.rst (and point to it here), since it is a bit non-obvious.

[kaushikcfd]

Use typing.cast in eliminate_dead_code.