NVIDIA / Fuser

A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
Other
271 stars 53 forks source link

Add `unsafeBind` to `LaunchParams` to allow modification. #3418

Closed rdspring1 closed 6 days ago

rdspring1 commented 6 days ago

The LaunchParams class can act as a scheduler parameter or an executor constraint. The bind function uses checkAndSet to enforce that the value for a ParallelType cannot be set multiple times. The only way to modify a LaunchParams is construct a copy and replace the desired value.

This PR adds bindUnsafe to LaunchParams that allows setting value of ParallelType multiple times. It is only for use when LaunchParams is a configuration parameter.

rdspring1 commented 6 days ago

!test

rdspring1 commented 6 days ago

!build