NjLSrSSKJr SSKJr SSKJr SSKJr SSK J r SSK r SS K J r S S K Jr /S Qr\"S5S \\\ 44Sj5rS \4SjrS \4SjrS \4SjrS \4SjrS \4SjrS \4SjrS \4SjrS \4SjrS&S\ RR8S S4Sjjr"SS5r"SS5r\"5r \ q!S&S\ RR8S \4Sjjr""SS\5r#S \S \4S!jr$S \%4S"jr&S\ RR8S S4S#jr'S \4S$jr(S \4S%jr)g)'zj This package implements abstractions found in ``torch.cuda`` to facilitate writing device-agnostic code. )Mapping)AbstractContextManager) lru_cache)MappingProxyType)AnyNdevice)amp) is_availableis_initialized synchronizecurrent_devicecurrent_streamstream set_device device_countStream StreamContextEventget_capabilitiesreturncd[[RRR 55$)ad Returns an immutable mapping of CPU capabilities detected at runtime. This function queries the CPU for supported instruction sets and features using cpuinfo. The result is cached after the first call for efficiency. The returned mapping contains architecture-specific capabilities: For x86/x86_64: - SSE family: sse, sse2, sse3, ssse3, sse4_1, sse4_2, sse4a - AVX family: avx, avx2, avx_vnni - AVX-512 family: avx512_f, avx512_cd, avx512_dq, avx512_bw, avx512_vl, avx512_ifma, avx512_vbmi, avx512_vbmi2, avx512_bitalg, avx512_vpopcntdq, avx512_vnni, avx512_bf16, avx512_fp16, avx512_vp2intersect, avx512_4vnniw, avx512_4fmaps - AVX10 family: avx10_1, avx10_2 - AVX-VNNI-INT: avx_vnni_int8, avx_vnni_int16, avx_ne_convert - AMX: amx_bf16, amx_tile, amx_int8, amx_fp16 - FMA: fma3, fma4 - Other: f16c, bmi, bmi2, popcnt, lzcnt, aes, sha, clflush, clflushopt, clwb For ARM64: - SIMD: neon, fp16_arith, bf16, i8mm, dot - SVE: sve, sve2, sve_bf16, sve_max_length (when supported) - SME: sme, sme2, sme_max_length (when supported) - Other: atomics, fhm, rdm, crc32, aes, sha1, sha2, pmull Common to all architectures: - architecture: string identifying the CPU architecture Returns: MappingProxyType: An immutable mapping where keys are capability names (e.g., 'avx2', 'sve') and values are booleans indicating support, or integers for properties like vector lengths. Example: >>> caps = torch.cpu.get_capabilities() >>> if caps.get("avx2", False): ... print("AVX2 is supported") >>> print(f"Architecture: {caps['architecture']}") )rtorch_C_cpu_get_cpu_capabilityc/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/torch/cpu/__init__.pyrr#s!V EHHMM==? @@r c6[5RSS5$)z/Returns a bool indicating if CPU supports AVX2.avx2Frgetrr r!_is_avx2_supportedr&Qs   ! !&% 00r c6[5RSS5$)z1Returns a bool indicating if CPU supports AVX512.avx512_fFr$rr r!_is_avx512_supportedr)V   ! !*e 44r c6[5RSS5$)z6Returns a bool indicating if CPU supports AVX512_BF16. avx512_bf16Fr$rr r!_is_avx512_bf16_supportedr-[s   ! !- 77r c6[5RSS5$)z/Returns a bool indicating if CPU supports VNNI. avx512_vnniFr$rr r!_is_vnni_supportedr0`s   ! !- 77r c6[5RSS5$)z3Returns a bool indicating if CPU supports AMX_TILE.amx_tileFr$rr r!_is_amx_tile_supportedr3fr*r c6[5RSS5$)z3Returns a bool indicating if CPU supports AMX FP16.amx_fp16Fr$rr r!_is_amx_fp16_supportedr6kr*r cR[RRR5$)zInitializes AMX instructions.)rrr _init_amxrr r!r8r8ps 88== " " $$r cg)z}Returns a bool indicating if CPU is currently available. N.B. This function only exists to facilitate device-agnostic code Trrr r!r r us r r cg)zWaits for all kernels in all streams on the CPU device to complete. Args: device (torch.device or int, optional): ignored, there's only one CPU device. N.B. This function only exists to facilitate device-agnostic code. Nrr s r!rr~r cL\rSrSrSrS S\SS4SjjrS SjrS SjrS S jr S r g) rz@ N.B. This class only exists to facilitate device-agnostic code priorityrNcgNr)selfr>s r!__init__Stream.__init__ r cgr@rrArs r! wait_streamStream.wait_streamrDr cgr@rrAs r! record_eventStream.record_eventrDr cgr@r)rAevents r! wait_eventStream.wait_eventrDr r)rN) __name__ __module__ __qualname____firstlineno____doc__intrBrGrKrO__static_attributes__rr r!rrs)  d    r rcH\rSrSrS\4SjrS S SjjrS SjrS S SjjrSr g) rrcg)NTrrJs r!query Event.querysr Ncgr@rrFs r!record Event.recordrDr cgr@rrJs r!rEvent.synchronizerDr cgr@rrFs r!wait Event.waitrDr rr@rR) rSrTrUrVboolr]r`rrerYrr r!rrs"t    r rc[$)zReturns the currently selected :class:`Stream` for a given device. Args: device (torch.device or int, optional): Ignored. N.B. This function only exists to facilitate device-agnostic code )_current_streamr s r!rrs  r cP\rSrSr%Sr\S-\S'SrSrS\ S\ S \ S S4S jr S r g) rznContext-manager that selects a given stream. N.B. This class only exists to facilitate device-agnostic code N cur_streamc&Xl[Ulgr@)r_default_cpu_stream prev_streamrFs r!rBStreamContext.__init__s .r c>URnUcg[UlUqgr@)rriro)rArls r! __enter__StreamContext.__enter__s#[[   +$r typevalue tracebackrc<URnUcgURqgr@)rrori)rArtrurvrls r!__exit__StreamContext.__exit__s![[   **r )ror) rSrTrUrVrWr__annotations__rBrrrrxrYrr r!rrs>  /%+S++++r rrc[U5$)zWrapper around the Context-manager StreamContext that selects a given stream. N.B. This function only exists to facilitate device-agnostic code )r)rs r!rrs   r cg)zxReturns number of CPU devices (not cores). Always 1. N.B. This function only exists to facilitate device-agnostic code r rrr r!rrs r cg)zrSets the current device, in CPU we do nothing. N.B. This function only exists to facilitate device-agnostic code Nrr s r!rrr;r cg)zqReturns current device for cpu. Always 'cpu'. N.B. This function only exists to facilitate device-agnostic code cpurrr r!rrs r cg)zxReturns True if the CPU is initialized. Always True. N.B. This function only exists to facilitate device-agnostic code Trrr r!rrs r r@)*rWcollections.abcr contextlibr functoolsrtypesrtypingrrr _devicer __all__strrrgr&r)r-r0r3r6r8r DevicerrrrnrirrrrXrrrrrr r!rsv $-"    4*A'#s(+*A*AZ1D1 5d5 848 8D8 55 55 %4% d **d  $   h% 5;;--  +*+>!6!4! c u{{))dr