Configuration for device settings.
| 874 | |
| 875 | |
| 876 | class DeviceConfig: |
| 877 | """ |
| 878 | Configuration for device settings. |
| 879 | """ |
| 880 | |
| 881 | def __init__( |
| 882 | self, |
| 883 | args, |
| 884 | ): |
| 885 | self.device_type = "cuda" |
| 886 | for key, value in args.items(): |
| 887 | if hasattr(self, key): |
| 888 | setattr(self, key, value) |
| 889 | |
| 890 | |
| 891 | class GraphOptimizationConfig: |
no outgoing calls