[mlir][Python] Fix comments of 'getCapsule' and 'createFromCapsule'

[zhanghb97]

Summary

The getCapsule and createFromCapsule comments incorrectly state the PyMlirContext and MlirContext in PyLocation, PyAttribute, and PyType classes.

[stellaraccident]

Thanks!

mlir/lib/Bindings/Python/IRModules.h

public:
   operator MlirLocation() const { return loc; }
   MlirLocation get() const { return loc; }
 
   /// Enter and exit the context manager.
   pybind11::object contextEnter();
   void contextExit(pybind11::object excType, pybind11::object excVal,
                    pybind11::object excTb);
 
-  /// Gets a capsule wrapping the void* within the MlirContext.
+  /// Gets a capsule wrapping the void* within the MlirLocation.
   pybind11::object getCapsule();
 
-  /// Creates a PyMlirContext from the MlirContext wrapped by a capsule.
-  /// Note that PyMlirContext instances are uniqued, so the returned object
-  /// may be a pre-existing object. Ownership of the underlying MlirContext
+  /// Creates a PyLocation from the MlirLocation wrapped by a capsule.
+  /// Note that PyLocation instances are uniqued, so the returned object
+  /// may be a pre-existing object. Ownership of the underlying MlirLocation
   /// is taken by calling this function.
   static PyLocation createFromCapsule(pybind11::object capsule);
 
 private:
   MlirLocation loc;
 };
 
 /// Used in function arguments when None should resolve to the current context

 class PyAttribute : public BaseContextObject {
 public:
   PyAttribute(PyMlirContextRef contextRef, MlirAttribute attr)
       : BaseContextObject(std::move(contextRef)), attr(attr) {}
   bool operator==(const PyAttribute &other);
   operator MlirAttribute() const { return attr; }
   MlirAttribute get() const { return attr; }
 
-  /// Gets a capsule wrapping the void* within the MlirContext.
+  /// Gets a capsule wrapping the void* within the MlirAttribute.
   pybind11::object getCapsule();
 
-  /// Creates a PyMlirContext from the MlirContext wrapped by a capsule.
-  /// Note that PyMlirContext instances are uniqued, so the returned object
-  /// may be a pre-existing object. Ownership of the underlying MlirContext
+  /// Creates a PyAttribute from the MlirAttribute wrapped by a capsule.
+  /// Note that PyAttribute instances are uniqued, so the returned object
+  /// may be a pre-existing object. Ownership of the underlying MlirAttribute
   /// is taken by calling this function.
   static PyAttribute createFromCapsule(pybind11::object capsule);
 
 private:
   MlirAttribute attr;
 };
 
 /// Represents a Python MlirNamedAttr, carrying an optional owned name.

 class PyType : public BaseContextObject {
 public:
   PyType(PyMlirContextRef contextRef, MlirType type)
       : BaseContextObject(std::move(contextRef)), type(type) {}
   bool operator==(const PyType &other);
   operator MlirType() const { return type; }
   MlirType get() const { return type; }
 
-  /// Gets a capsule wrapping the void* within the MlirContext.
+  /// Gets a capsule wrapping the void* within the MlirType.
   pybind11::object getCapsule();
 
-  /// Creates a PyMlirContext from the MlirContext wrapped by a capsule.
-  /// Note that PyMlirContext instances are uniqued, so the returned object
-  /// may be a pre-existing object. Ownership of the underlying MlirContext
+  /// Creates a PyType from the MlirType wrapped by a capsule.
+  /// Note that PyType instances are uniqued, so the returned object
+  /// may be a pre-existing object. Ownership of the underlying MlirType
   /// is taken by calling this function.
   static PyType createFromCapsule(pybind11::object capsule);
 
 private:
   MlirType type;
 };
 
 /// Wrapper around the generic MlirValue.