Add types to elementwise.py

[alexfikl]

This adds types pretty much all things in elementwise.py. There are some remaining errors that I'm not sure how to best handle:

1. context_dependent_memoize expects something hashable as a first argument, mostly due to this:
   

   pyopencl/pyopencl/tools.py

   Line 583 in 2acc9f7

   return cache_dict[cl_object][args]

However, the cl.Context that the functions in elementwise use is very much not hashable (?).

2. Added a debatable ScalarLike alias for some functions in array.py.
3. Deprecated range and slice in ElementwiseKernel and used named that don't shadow global functions instead. Went for this so that their types are not hidden in the kwargs.

[inducer]

However, the cl.Context that the functions in elementwise use is very much not hashable (?).

🤔 What makes you say that? Context is supposed to be hashable.

pyopencl/pyopencl/_cl.pyi

Lines 1004 to 1005 in 2acc9f7

	@override
	def __hash__(self) -> int: ...

[alexfikl]

Hm, mostly an unwavering trust in pyright:

pyopencl/pyopencl/elementwise.py:483:2 - error: Argument of type "(context: Context, dtype: dtype[Any], idx_dtype: dtype[Any], vec_count: int = 1) -> Kernel" cannot be assigned to parameter "func" of type "(Hashable, **P@first_arg_dependent_memoize) -> RetT@first_arg_dependent_memoize" in function "first_arg_dependent_memoize"
Type "(context: Context, dtype: dtype[Any], idx_dtype: dtype[Any], vec_count: int = 1) -> Kernel" is not assignable to type "(Hashable, **P@first_arg_dependent_memoize)-> RetT@first_arg_dependent_memoize"
Parameter 1: type "Hashable" is incompatible with type "Context"
"Hashable" is not assignable to "Context" (reportArgumentType)

Not quite sure what it doesn't like then, since it should see the __hash__ override. Any ideas?

[inducer]

Thx! Only two minor nits.

[inducer]

Not quite sure what it doesn't like then, since it should see the __hash__ override. Any ideas?

Note that this goes the other way! "Hashable" is not assignable to "Context", not "Context" is not assignable to "Hashable"

[alexfikl]

Note that this goes the other way! "Hashable" is not assignable to "Context", not "Context" is not assignable to "Hashable"

🤦 It makes a lot more sense when you actually read it, yeah :( Should be fixed now!

[inducer]

Thx!