flexflow · lambda7xx · Oct 5, 2023 · Oct 7, 2023 · Oct 8, 2023 · Oct 8, 2023
diff --git a/deps/fmt b/deps/fmt
diff --git a/lib/kernels/src/cuda/aggregate_spec_kernels.cu b/lib/kernels/src/cuda/aggregate_spec_kernels.cu
diff --git a/lib/op-attrs/include/op-attrs/ops/attention.h b/lib/op-attrs/include/op-attrs/ops/attention.h
@@ -12,6 +12,7 @@ struct MultiHeadAttentionAttrs {
   req<float> dropout;
   req<bool> bias, add_bias_kv, add_zero_attn;
 };
+
 FF_VISITABLE_STRUCT(MultiHeadAttentionAttrs,
                     embed_dim,
                     num_heads,

diff --git a/lib/op-attrs/include/op-attrs/ops/batch_matmul.h b/lib/op-attrs/include/op-attrs/ops/batch_matmul.h
@@ -14,6 +14,9 @@ FF_VISITABLE_STRUCT(BatchMatmulAttrs, a_seq_length_dim, b_seq_length_dim);
 
 CHECK_VALID_OP_ATTR(BatchMatmulAttrs);
 
+ParallelTensorShape get_output_shape(BatchMatmulAttrs const &,
+                                     ParallelTensorShape const &,
+                                     ParallelTensorShape const &);
 } // namespace FlexFlow
 
 #endif
diff --git a/lib/op-attrs/include/op-attrs/ops/batch_norm.h b/lib/op-attrs/include/op-attrs/ops/batch_norm.h
@@ -12,7 +12,8 @@ struct BatchNormAttrs {
 };
 FF_VISITABLE_STRUCT(BatchNormAttrs, relu);
 
-ParallelTensorShape get_output_shape(BatchNormAttrs const &);
+ParallelTensorShape get_output_shape(BatchNormAttrs const &,
+                                     ParallelTensorShape const &);
 
 CHECK_VALID_OP_ATTR(BatchNormAttrs);
 

diff --git a/lib/op-attrs/include/op-attrs/ops/cast.h b/lib/op-attrs/include/op-attrs/ops/cast.h
@@ -13,6 +13,9 @@ struct CastAttrs {
 };
 FF_VISITABLE_STRUCT(CastAttrs, dtype);
 
+ParallelTensorShape get_output_shape(CastAttrs const &,
+                                     ParallelTensorShape const &);
+
 CHECK_VALID_OP_ATTR(CastAttrs);
 } // namespace FlexFlow
 

diff --git a/lib/op-attrs/include/op-attrs/ops/combine.h b/lib/op-attrs/include/op-attrs/ops/combine.h
@@ -15,6 +15,9 @@ struct CombineAttrs {
 FF_VISITABLE_STRUCT(CombineAttrs, combine_dim, combine_degree);
 CHECK_VALID_OP_ATTR(CombineAttrs);
 
+ParallelTensorShape get_output_shape(CombineAttrs const &,
+                                     ParallelTensorShape const &);
+
 } // namespace FlexFlow
 
 #endif
diff --git a/lib/op-attrs/include/op-attrs/ops/concat.h b/lib/op-attrs/include/op-attrs/ops/concat.h
@@ -15,6 +15,8 @@ struct ConcatAttrs {
 FF_VISITABLE_STRUCT(ConcatAttrs, axis, num_inputs);
 CHECK_VALID_OP_ATTR(ConcatAttrs);
 
+ParallelTensorShape get_output_shape(ConcatAttrs const &,
+                                     std::vector<ParallelTensorShape> const &);
 } // namespace FlexFlow
 
 #endif
diff --git a/lib/op-attrs/include/op-attrs/ops/conv_2d.h b/lib/op-attrs/include/op-attrs/ops/conv_2d.h
@@ -32,6 +32,9 @@ CHECK_VALID_OP_ATTR(Conv2DAttrs);
 TensorShape get_kernel_shape(Conv2DAttrs const &, TensorShape const &);
 TensorShape get_bias_shape(Conv2DAttrs const &, TensorShape const &);
 
+ParallelTensorShape get_output_shape(Conv2DAttrs const &,
+                                     ParallelTensorShape const &);
+
 } // namespace FlexFlow
 
 #endif
diff --git a/lib/op-attrs/include/op-attrs/ops/dropout.h b/lib/op-attrs/include/op-attrs/ops/dropout.h
@@ -14,6 +14,9 @@ struct DropoutAttrs {
 FF_VISITABLE_STRUCT(DropoutAttrs, rate, seed);
 CHECK_VALID_OP_ATTR(DropoutAttrs);
 
+ParallelTensorShape get_output_shape(DropoutAttrs const &,
+                                     ParallelTensorShape const &);
+
 } // namespace FlexFlow
 
 #endif
+0 −8		.github/dependabot.yml
+2 −3		.github/pull_request_template.md
+0 −30		.github/workflows/cifuzz.yml
+1 −9		.github/workflows/doc.yml
+13 −30		.github/workflows/linux.yml
+2 −17		.github/workflows/macos.yml
+0 −65		.github/workflows/scorecard.yml
+4 −12		.github/workflows/windows.yml
+41 −98		CMakeLists.txt
+1 −668		ChangeLog.rst
+1 −1		LICENSE.rst
+28 −42		README.rst
+145 −200		doc/api.rst
+6 −5		doc/build.py
+8 −134		doc/syntax.rst
+1 −1		include/fmt/args.h
+327 −466		include/fmt/chrono.h
+37 −18		include/fmt/color.h
+104 −27		include/fmt/compile.h
+1,139 −738		include/fmt/core.h
+152 −91		include/fmt/format-inl.h
+788 −1,081		include/fmt/format.h
+71 −44		include/fmt/os.h
+50 −22		include/fmt/ostream.h
+174 −201		include/fmt/printf.h
+145 −158		include/fmt/ranges.h
+37 −331		include/fmt/std.h
+33 −62		include/fmt/xchar.h
+29 −40		src/fmt.cc
+5 −1		src/format.cc
+62 −99		src/os.cc
+3 −3		support/Vagrantfile
+1 −0		support/bazel/.bazelrc
+1 −1		support/bazel/.bazelversion
+2 −2		support/bazel/BUILD.bazel
+4 −5		support/bazel/README.md
+1 −1		support/build.gradle
+54 −0		support/cmake/cxx14.cmake
+0 −7		support/rst2md.py
+3 −15		test/CMakeLists.txt
+1 −1		test/add-subdirectory-test/CMakeLists.txt
+1 −1		test/args-test.cc
+26 −386		test/chrono-test.cc
+2 −2		test/compile-error-test/CMakeLists.txt
+0 −1		test/compile-fp-test.cc
+40 −22		test/compile-test.cc
+239 −119		test/core-test.cc
+0 −2		test/enforce-checks-test.cc
+1 −1		test/find-package-test/CMakeLists.txt
+91 −27		test/format-impl-test.cc
+329 −285		test/format-test.cc
+1 −1		test/fuzzing/CMakeLists.txt
+0 −2		test/gtest-extra-test.cc
+6 −1		test/gtest-extra.h
+7 −1		test/gtest/CMakeLists.txt
+1 −1		test/gtest/gmock-gtest-all.cc
+2 −2		test/mock-allocator.h
+88 −36		test/module-test.cc
+56 −15		test/os-test.cc
+47 −16		test/ostream-test.cc
+7 −1		test/posix-mock-test.cc
+2 −0		test/posix-mock.h
+37 −13		test/printf-test.cc
+35 −138		test/ranges-test.cc
+1 −1		test/scan-test.cc
+14 −17		test/scan.h
+1 −1		test/static-export-test/CMakeLists.txt
+20 −198		test/std-test.cc
+6 −2		test/util.h
+55 −118		test/xchar-test.cc