good first issue
Métricas do repositório
- Stars
- (392 stars)
- Métricas de merge de PR
- (Métricas PR pendentes)
Description
We need to update the python benchmarks to use the direct bindings. I'd recommend updating the test files that only need updates to their FusionDefinition before updating core.py.
- CI command:
!test --pybench
General
- Replace
import nvfuserwithimport nvfuser_direct as nvfuser
Update core.py in benchmarks/python
- Direct bindings does not use
FusionCache. Remove frombenchmarks/python/core.pyand update benchmarks appropriately. - Direct bindings does not support
profileargument infd.execute.profilingwas disabled for CUDA 13. Either add profiling to direct bindings or remove option from benchmarks. benchmarks/python/hosttests theFusionCache, so you can make a copy ofcore.pyjust for these tests OR remove the completely.
Update FusionDefinition for benchmarks.
There are some API Differences between direct and legacy bindings. Update FusionDefinition so they run correctly.
- Replace
fd.define_vectorwith regular python list.
V6 = fd.define_vector([T0.size(0), 1], dtype=DataType.Int) # Not Implemented in direct bindings
V6 = [T0.size(0), 1] # Replace with regular python list
- Replace
fd.add_output(output, stride_order)with
output = fd.stride_order(a, stride_order)
fd.add_output(output)