Explicitly disallow sparse dots in elemental emitter and cublas
Also fix assertions in passes that verify number of dot operands (sparse dot has a 3rd metadata operand). PiperOrigin-RevId: 615382282
Showing
- tensorflow/core/public/version.h 1 addition, 1 deletiontensorflow/core/public/version.h
- third_party/xla/xla/service/BUILD 9 additions, 0 deletionsthird_party/xla/xla/service/BUILD
- third_party/xla/xla/service/elemental_ir_emitter.cc 4 additions, 0 deletionsthird_party/xla/xla/service/elemental_ir_emitter.cc
- third_party/xla/xla/service/elemental_ir_emitter.h 25 additions, 0 deletionsthird_party/xla/xla/service/elemental_ir_emitter.h
- third_party/xla/xla/service/elemental_ir_emitter_test.cc 46 additions, 4 deletionsthird_party/xla/xla/service/elemental_ir_emitter_test.cc
- third_party/xla/xla/service/gpu/BUILD 1 addition, 0 deletionsthird_party/xla/xla/service/gpu/BUILD
- third_party/xla/xla/service/gpu/gemm_rewriter.cc 5 additions, 0 deletionsthird_party/xla/xla/service/gpu/gemm_rewriter.cc
- third_party/xla/xla/service/gpu/gemm_rewriter_test.cc 18 additions, 0 deletionsthird_party/xla/xla/service/gpu/gemm_rewriter_test.cc
- third_party/xla/xla/service/gpu/gpu_float_support.cc 2 additions, 1 deletionthird_party/xla/xla/service/gpu/gpu_float_support.cc
- third_party/xla/xla/service/gpu/ir_emission_utils.cc 2 additions, 22 deletionsthird_party/xla/xla/service/gpu/ir_emission_utils.cc
Please register or sign in to comment