search

openxla / stablehlo

Backward compatible ML compute opset inspired by HLO/MHLO
Apache License 2.0
357 stars 100 forks source link

Adding multi-operation quantization tests #2413

Closed sdasgup3 closed 3 days ago

sdasgup3 commented 1 week ago

Adds multi-operation quantized test cases

There aren't many multi-operation testdata files in https://github.com/openxla/stablehlo/tree/main/stablehlo/testdata satisfying

  1. The file should cover ops from the list cbrt, ceil, cosine, divide, exponential, exponential_minus_one, floor, log, log_plus_one, logistic, multiply, negate, sign, sine, sqrt, subtract, tanh,
  2. The file should test only f32 type

Going forward will will be expanding on the op/type coverage.