Add a feature flag for nonscalar broadcasting

arraycontext/context.py

@@ -355,6 +355,11 @@ class ArrayContext(ABC):
     def permits_inplace_modification(self):
         pass
 
+    # undocumented for now
+    @abstractproperty
+    def supports_nonscalar_broadcasting(self):
+        pass
+
 # }}}
 
 # vim: foldmethod=marker

arraycontext/impl/pyopencl/__init__.py

@@ -278,6 +278,10 @@ class PyOpenCLArrayContext(ArrayContext):
     def permits_inplace_modification(self):
         return True
 
+    @property
+    def supports_nonscalar_broadcasting(self):
+        return False
+
 # }}}
 
 # vim: foldmethod=marker

arraycontext/impl/pytato/__init__.py

@@ -227,3 +227,7 @@ class PytatoPyOpenCLArrayContext(ArrayContext):
     @property
     def permits_inplace_modification(self):
         return False
+
+    @property
+    def supports_nonscalar_broadcasting(self):
+        return True