comments

Author: rdspring1

doc/dev/python_scheduling/autotune_inner_reduction.py

@@ -5,6 +5,8 @@
 
 import torch
 import itertools
+
+# TODO Update script to use nvfuser_direct module
 from nvfuser import FusionDefinition, SchedulerType, DataType, ParallelType
 from enum import Enum
 from dataclasses import dataclass

doc/dev/python_scheduling/autotune_matmul.py

@@ -5,6 +5,8 @@
 
 import torch
 import itertools
+
+# TODO Update script to use nvfuser_direct module
 from nvfuser import FusionDefinition, SchedulerType
 
 # Description of the problem

doc/dev/python_scheduling/autotune_pointwise.py

@@ -6,6 +6,8 @@
 import torch
 import itertools
 import math
+
+# TODO Update script to use nvfuser_direct module
 from nvfuser import FusionDefinition, SchedulerType, DataType
 from dataclasses import dataclass
 from enum import Enum

doc/dev/python_scheduling/profile_matmul.py

@@ -1,6 +1,7 @@
 # SPDX-FileCopyrightText: Copyright (c) 2025-present NVIDIA CORPORATION & AFFILIATES.
 # All rights reserved.
 # SPDX-License-Identifier: BSD-3-Clause
+# TODO Update script to use nvfuser_direct module
 from nvfuser import (
     FusionDefinition,
     SchedulerType,

python/utils.py

@@ -547,8 +547,7 @@ def run(config, version_tag, relative_path):
     from setuptools import Extension, setup, find_packages
 
     # NOTE(crcrpar): Deliberately build basically two dynamic libraries here so that they can
-    # be treated as "nvfuser_package_data". This function call will put the two of "nvfuser" and
-    # "nvfuser_codegen" into "./nvfuser/lib", and the former will be "nvfuser._C".
+    # be treated as "nvfuser_package_data".
     if config.build_setup:
         cmake(config, relative_path)
     if not config.cmake_only: