    def __init__(
        self,
    ) -> None:
        """initialize the op"""
        super().__init__()
        self._attrs["op"] = "softmax"
        self._attrs["has_profiler"] = False
        if detect_target().name() == "rocm":
            self._attrs["has_profiler"] = True
initialize the op
python/aitemplate/compiler/ops/softmax/softmax.py