good first issue
倉庫指標
- Star
- (392 star)
- PR 合併指標
- (PR 指標待抓取)
描述
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)