o Ti @s ddlmZGdddeZdS)) CUDAOpBuildercsFeZdZdZdZdfdd ZddZdd Zd d Zd d Z Z S)QuantizerBuilderDS_BUILD_QUANTIZER quantizerNcs$|dur|jn|}tj|ddS)N)name)NAMEsuper__init__)selfr __class__V/home/ubuntu/.local/lib/python3.10/site-packages/deepspeed/ops/op_builder/quantizer.pyr szQuantizerBuilder.__init__cCsd|jdS)Nzdeepspeed.ops.quantizer._op)rr r r r absolute_nameszQuantizerBuilder.absolute_namecCsgdS)N)z csrc/quantization/pt_binding.cppz#csrc/quantization/fake_quantizer.cuzcsrc/quantization/quantize.cuz"csrc/quantization/quantize_intX.cuzcsrc/quantization/dequantize.cuz&csrc/quantization/swizzled_quantize.cuz!csrc/quantization/quant_reduce.cur rr r rsourcesszQuantizerBuilder.sourcescCsdgS)Nz csrc/includesr rr r r include_pathsszQuantizerBuilder.include_pathscCs|sdgSgS)Nz-lcurand)is_rocm_pytorchrr r r extra_ldflags"szQuantizerBuilder.extra_ldflags)N) __name__ __module__ __qualname__ BUILD_VARrr rrrr __classcell__r r r rr s rN)builderrrr r r rs