saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-05 04:17:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[FIX] reorder initializer (#18097)

Browse source 
### Description
Fix building error when with collective ops: error is thrown because
`device_mesh_axis` will be initialized after `cond`.
This commit is contained in:
mindest 2023-10-27 11:29:55 +08:00 committed by GitHub
parent b7408f7389
commit 0f3a067d3a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/contrib_ops/cuda/collective/sharding_spec.h
View file

@ -117,7 +117,7 @@ class AxisPartitionSpec {
// A normal ctor.
// TODO(wechi): Consider to hide it and revise the `public` members/functions
// exposed to the user.
AxisPartitionSpec(Condition cond_, int device_mesh_axis_) : device_mesh_axis(device_mesh_axis_), cond(cond_) {}
AxisPartitionSpec(Condition cond_, int device_mesh_axis_) : cond(cond_), device_mesh_axis(device_mesh_axis_) {}
// Helper to debug and generate error message; e.g.,
// "RS[0]".