Method run

python/setup.py:619–639

(self)

        subprocess.check_call(["make_maca"] + build_args, cwd=cmake_dir, env=env)

    def run(self):
        # We creates symbolic links for each file in backend separately. Since the nvshmem bitcode is moved to nvidia/lib,
        # it needs to be built first.
        build_shmem()
        for ext in self.extensions:
            if isinstance(ext, CMakeExtension):
                self.build_extension_cmake(ext)

            if _is_maca_platform() and check_env_flag("TRITON_USE_MACA", "OFF"):
                pymxshmem_ext_dir = get_pymxshmem_project()
                print(f"build pymxshmem:{pymxshmem_ext_dir}", file=sys.stderr)
                self.build_pymxshmem_project(ext, pymxshmem_ext_dir)

                pymaca_project = get_pymaca_project()
                print(f"build pymaca:{pymaca_project}", file=sys.stderr)
                self.build_pymaca_project(ext, pymaca_project)

        all_extensions = self.extensions
        self.extensions = [ext for ext in self.extensions if not isinstance(ext, CMakeExtension)]
        super().run()
        self.extensions = all_extensions

    def get_pybind11_cmake_args(self):
        pybind11_sys_path = get_env_with_keys(["PYBIND11_SYSPATH"])