NFC: Rename `DispatchDefaultWorkgroupCountOp` to `DispatchWorkgroupCountFromDagRootOp` (#10344)
The renaming provides a bit more context on computation the operation represents when used at the Flow level as a placeholder for returning the number of workgroups. The documentation of the operation and the name are updated as well.
diff --git a/tests/compiler_driver/streams.mlir b/tests/compiler_driver/streams.mlir
index e75ba36..9de9f1c 100644
--- a/tests/compiler_driver/streams.mlir
+++ b/tests/compiler_driver/streams.mlir
@@ -29,7 +29,7 @@
// CHECK: vm.rodata private @executable_0_vmvx_bytecode_fb
stream.executable private @executable_0 {
stream.executable.export public @dispatch workgroups(%arg0: index) -> (index, index, index) {
- %x, %y, %z = flow.dispatch.default_workgroup_count %arg0
+ %x, %y, %z = flow.dispatch.workgroup_count_from_dag_root %arg0
stream.return %x, %y, %z : index, index, index
}
builtin.module {
@@ -77,7 +77,7 @@
// CHECK: vm.rodata private @executable_1_vmvx_bytecode_fb
stream.executable private @executable_1 {
stream.executable.export public @dispatch workgroups(%arg0: index) -> (index, index, index) {
- %x, %y, %z = flow.dispatch.default_workgroup_count %arg0
+ %x, %y, %z = flow.dispatch.workgroup_count_from_dag_root %arg0
stream.return %x, %y, %z : index, index, index
}
builtin.module {
@@ -126,7 +126,7 @@
// CHECK: vm.rodata private @executable_2_vmvx_bytecode_fb
stream.executable private @executable_2 {
stream.executable.export public @dispatch workgroups(%arg0: index) -> (index, index, index) {
- %x, %y, %z = flow.dispatch.default_workgroup_count %arg0
+ %x, %y, %z = flow.dispatch.workgroup_count_from_dag_root %arg0
stream.return %x, %y, %z : index, index, index
}
builtin.module {
