Merged
turbo_quant_compress_latent csrc upgrade #8607
丛吉钰Add on Aug 12
turbo_quant_compress_latent csrc upgrade #8607
Merged
Merged fromcong-jiyu/ops-nntorchintocann/ops-nnmaster
Total: 1 files changed+2-2
| @@ -34,10 +34,10 @@ class TurboQuantCompressLatentOpBuilder(OpBuilder): | |||
| 34 | """ | 34 | """ |
| 35 | 35 | ||
| 36 | def __init__(self): | 36 | def __init__(self): |
| 37 | - super().__init__("turbo_quant_compress_latent", "quant") | 37 | + super().__init__("turbo_quant_compress_latent") |
| 38 | 38 | ||
| 39 | def sources(self) -> list: | 39 | def sources(self) -> list: |
| 40 | - return ["csrc/quant/turbo_quant_compress_latent.cpp"] | 40 | + return [self.resolve_source("turbo_quant_compress_latent.cpp")] |
| 41 | 41 | ||
| 42 | def schema(self) -> str: | 42 | def schema(self) -> str: |
| 43 | return "turbo_quant_compress_latent(Tensor latent, Tensor centroids) -> Tensor" | 43 | return "turbo_quant_compress_latent(Tensor latent, Tensor centroids) -> Tensor" |